ki-backend/ki/models.py

98 lines
3.1 KiB
Python

from datetime import datetime
from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Table
from sqlalchemy.orm import relationship
from app import db
user_skill_table = Table("user_skill", db.Model.metadata,
Column("user_id", Integer, ForeignKey("user.id")),
Column("skill_id", Integer, ForeignKey("skill.id")))
user_language_table = Table(
"user_language", db.Model.metadata,
Column("user_id", Integer, ForeignKey("user.id")),
Column("language_id", String(2), ForeignKey("language.id")))
class User(db.Model):
__tablename__ = "user"
id = Column(Integer, primary_key=True)
nickname = Column(String(25), unique=True, nullable=False)
pronouns = Column(String(25), default="")
volunteerwork = Column(String(4000), default="")
freetext = Column(String(4000), default="")
created = Column(DateTime, nullable=False)
updated = Column(DateTime, onupdate=datetime.now, nullable=False)
contacts = relationship("Contact")
address = relationship("Address", uselist=False, back_populates="user")
skills = relationship("Skill",
secondary=user_skill_table,
back_populates="users")
languages = relationship("Language",
secondary=user_language_table,
back_populates="users")
class Contact(db.Model):
__tablename__ = "contact"
id = Column(Integer, primary_key=True)
user_id = Column(Integer, ForeignKey("user.id"))
user = relationship("User", back_populates="contacts")
contacttype_id = Column(Integer, ForeignKey("contacttype.id"))
contacttype = relationship("ContactType")
content = Column(String(200), nullable=False)
class ContactType(db.Model):
__tablename__ = "contacttype"
id = Column(Integer, primary_key=True)
name = Column(String(25), nullable=False)
class Address(db.Model):
__tablename__ = "address"
id = Column(Integer, primary_key=True)
street = Column(String(25), default="")
house_number = Column(String(10), default="")
additional = Column(String(25), default="")
postcode = Column(String(10), default="")
city = Column(String(25), default="")
country = Column(String(25), default="")
user_id = Column(Integer, ForeignKey("user.id"))
user = relationship("User", back_populates="address")
class Skill(db.Model):
__tablename__ = "skill"
id = Column(Integer, primary_key=True)
name = Column(String(25), unique=True, nullable=False)
users = relationship("User",
secondary=user_skill_table,
back_populates="skills")
def to_dict(self):
return {"id": self.id, "name": self.name}
class Language(db.Model):
__tablename__ = "language"
id = Column(String(2), primary_key=True)
name = Column(String(25), nullable=False)
users = relationship("User",
secondary=user_language_table,
back_populates="languages")
def to_dict(self):
return {"id": self.id, "name": self.name}