forked from s_ranjbar/city_retrofit
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
"""
|
|
Model representation of a User
|
|
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
|
Copyright © 2022 Concordia CERC group
|
|
Project Coder Peter Yefi peteryefi@gmail.com
|
|
"""
|
|
|
|
from sqlalchemy import Column, Integer, String, Sequence
|
|
from sqlalchemy import DateTime, Enum
|
|
from hub.persistence.db_config import Base
|
|
import datetime
|
|
from sqlalchemy.orm import validates
|
|
import re
|
|
import enum
|
|
from sqlalchemy.orm import relationship
|
|
|
|
|
|
class UserRoles(enum.Enum):
|
|
Admin = 'Admin'
|
|
Hub_Reader = 'Hub_Reader'
|
|
|
|
|
|
class User(Base):
|
|
"""A model representation of a city
|
|
"""
|
|
__tablename__ = "user"
|
|
id = Column(Integer, Sequence('user_id_seq'), primary_key=True)
|
|
name = Column(String, nullable=False)
|
|
password = Column(String, nullable=False)
|
|
role = Column(Enum(UserRoles), nullable=False, default=UserRoles.Hub_Reader)
|
|
application_id = Column(Integer, nullable=False)
|
|
created = Column(DateTime, default=datetime.datetime.utcnow)
|
|
updated = Column(DateTime, default=datetime.datetime.utcnow)
|
|
|
|
def __init__(self, name, password, role, application_id):
|
|
self.name = name
|
|
self.password = password
|
|
self.role = role
|
|
self.application_id = application_id
|