system_assignation/hub/persistence/models/user.py

"""
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)
  email = Column(String, nullable=False, unique=True)
  password = Column(String, nullable=False)
  role = Column(Enum(UserRoles), nullable=False, default=UserRoles.Hub_Reader)
  cities = relationship("City", back_populates="user")
  created = Column(DateTime, default=datetime.datetime.utcnow)
  updated = Column(DateTime, default=datetime.datetime.utcnow)

  @validates("email")
  def validate_email(self, key, address):
    pattern = r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b'
    if not re.match(pattern, address):
      raise ValueError("failed simple email validation")
    return address

  def __init__(self, name, email, password, role):
    self.name = name
    self.email = email
    self.password = password
    self.role = role