pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add lots of NOT NULL constraints

Browse Source
This commit is contained in:
richard-dds
2018-09-21 13:54:34 -04:00
parent 311fb34ac9
commit de338f675f
9 changed files with 34 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
atst/models/attachment.py
View File

@@ -14,8 +14,8 @@ class Attachment(Base, mixins.TimestampsMixin):
__tablename__ = "attachments"
id = types.Id()
filename = Column(String)
object_name = Column(String, unique=True)
filename = Column(String, nullable=False)
object_name = Column(String, unique=True, nullable=False)
@classmethod
def attach(cls, fyle):

2
atst/models/environment.py
View File

@@ -12,7 +12,7 @@ class Environment(Base, TimestampsMixin):
id = Id()
name = Column(String, nullable=False)
project_id = Column(ForeignKey("projects.id"))
project_id = Column(ForeignKey("projects.id"), nullable=False)
project = relationship("Project")
@property

6
atst/models/environment_role.py
View File

@@ -14,12 +14,14 @@ class EnvironmentRole(Base, mixins.TimestampsMixin):
__tablename__ = "environment_roles"
id = types.Id()
environment_id = Column(UUID(as_uuid=True), ForeignKey("environments.id"))
environment_id = Column(
UUID(as_uuid=True), ForeignKey("environments.id"), nullable=False
)
environment = relationship("Environment", backref="roles")
role = Column(String())
user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"))
user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"), nullable=False)
user = relationship("User", backref="environment_roles")

4
atst/models/request_internal_comment.py
View File

@@ -8,9 +8,9 @@ class RequestInternalComment(Base, mixins.TimestampsMixin):
__tablename__ = "request_internal_comments"
id = types.Id()
text = Column(String())
text = Column(String(), nullable=False)
user_id = Column(ForeignKey("users.id"), nullable=False)
user = relationship("User")
request_id = Column(ForeignKey("requests.id", ondelete="CASCADE"))
request_id = Column(ForeignKey("requests.id", ondelete="CASCADE"), nullable=False)

4
atst/models/request_status_event.py
View File

@@ -29,7 +29,9 @@ class RequestStatusEvent(Base, mixins.TimestampsMixin):
new_status = Column(SQLAEnum(RequestStatus, native_enum=False))
time_created = Column(DateTime(timezone=True), server_default=func.now())
request_id = Column(
UUID(as_uuid=True), ForeignKey("requests.id", ondelete="CASCADE")
UUID(as_uuid=True),
ForeignKey("requests.id", ondelete="CASCADE"),
nullable=False,
)
sequence = Column(
BigInteger, Sequence("request_status_events_sequence_seq"), nullable=False

6
atst/models/role.py
View File

@@ -9,9 +9,9 @@ class Role(Base, mixins.TimestampsMixin):
__tablename__ = "roles"
id = types.Id()
name = Column(String, index=True, unique=True)
description = Column(String)
permissions = Column(ARRAY(String), index=True, server_default="{}")
name = Column(String, index=True, unique=True, nullable=False)
description = Column(String, nullable=False)
permissions = Column(ARRAY(String), index=True, server_default="{}", nullable=False)
def add_permission(self, permission):
perms_set = set(self.permissions)

2
atst/models/user.py
View File

@@ -17,7 +17,7 @@ class User(Base, mixins.TimestampsMixin):
workspace_roles = relationship("WorkspaceRole", backref="user")
email = Column(String, unique=True)
dod_id = Column(String, unique=True)
dod_id = Column(String, unique=True, nullable=False)
first_name = Column(String)
last_name = Column(String)

6
atst/models/workspace_role.py
View File

@@ -10,13 +10,13 @@ class WorkspaceRole(Base, mixins.TimestampsMixin):
__tablename__ = "workspace_roles"
id = Id()
workspace_id = Column(UUID(as_uuid=True), ForeignKey("workspaces.id"), index=True)
workspace_id = Column(UUID(as_uuid=True), ForeignKey("workspaces.id"), index=True, nullable=False)
workspace = relationship("Workspace", back_populates="roles")
role_id = Column(UUID(as_uuid=True), ForeignKey("roles.id"))
role_id = Column(UUID(as_uuid=True), ForeignKey("roles.id"), nullable=False)
role = relationship("Role")
user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"), index=True)
user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"), index=True, nullable=False)
Index(