mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-31 23:26:23 -05:00
Create a relationship between services and tokens
This commit is contained in:
@@ -7,6 +7,7 @@ from app.models import (Service, User)
|
||||
|
||||
|
||||
def save_model_service(service, update_dict=None):
|
||||
print(update_dict)
|
||||
users_list = update_dict.get('users', []) if update_dict else getattr(service, 'users', [])
|
||||
if not users_list:
|
||||
error_msg = {'users': ['Missing data for required attribute']}
|
||||
|
||||
@@ -65,8 +65,9 @@ class Token(db.Model):
|
||||
__tablename__ = 'tokens'
|
||||
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
token = db.Column(db.String, unique=True, nullable=False)
|
||||
token = db.Column(db.String(255), unique=True, nullable=False)
|
||||
service_id = db.Column(db.Integer, db.ForeignKey('services.id'), index=True, nullable=False)
|
||||
service = db.relationship('Service', backref=db.backref('tokens', lazy='dynamic'))
|
||||
expiry_date = db.Column(db.DateTime)
|
||||
|
||||
|
||||
|
||||
@@ -19,7 +19,7 @@ class UserSchema(ma.ModelSchema):
|
||||
class ServiceSchema(ma.ModelSchema):
|
||||
class Meta:
|
||||
model = models.Service
|
||||
exclude = ("updated_at", "created_at", "templates")
|
||||
exclude = ("updated_at", "created_at", "tokens", "templates")
|
||||
|
||||
|
||||
class TemplateSchema(ma.ModelSchema):
|
||||
|
||||
Reference in New Issue
Block a user