class Profile(UserMixin, CRUDMixin, db.Model):
__tablename__ = "users_profile"
id = db.Column(db.Integer, primary_key=True)
UserID = db.Column(db.Integer, db.ForeignKey('users_user.id'))
user = db.relationship("User", back_populates='profile')
wardrobe = db.relationship('Wardrobe', back_populates='profile')
height = db.Column(db.Integer)
weight = db.Column(db.Integer)
build = db.Column(db.String(120))
shape = db.Column(db.String(120))
birthdate = db.Column(db.Date)
def get_age(self):
today = date.today()
born = self.birthdate
return today.year - born.year - ((today.month, today.day) <
(born.month, born.day))
def __iter__(self):
yield 'height', self.height
yield 'weidght', self.weight
yield 'build', self.build
yield 'shape', self.shape
yield 'birthdate', self.birthdate
def __repr__(self):
return f'<Profile ID: {self.id}>'
class Upper(CRUDMixin, db.Model):
__tablename__ = "wardrobe_upper"
id = db.Column(db.Integer, primary_key=True)
WardrobeID = db.Column(db.Integer, db.ForeignKey('wardrobe.id'))
wardrobe = db.relationship("Wardrobe", back_populates='uppers')
brand = db.Column(db.String(120))
style = db.Column(db.String(120))
collar = db.Column(db.Boolean)
size = db.Column(db.String(10))
year = db.Column(db.Integer)
pattern = db.Column(db.String(120))
color_1 = db.Column(db.String(120))
color_2 = db.Column(db.String(120))
color_3 = db.Column(db.String(120))
fit = db.Column(db.String(120))
material = db.Column(db.String(120))
stretch = db.Column(db.Boolean)
def __repr__(self):
return f"<Upper ID: {self.id}>"
def __iter__(self):
yield 'brand', self.brand
yield 'style', self.style
yield 'collar', self.collar
yield 'size', self.size
yield 'year', self.year
yield 'pattern', self.pattern
yield 'colors:', [self.color1, self.color2, self.color3]
yield 'fit', self.fit
yield 'material', self.material
yield 'stretch', self.stretch
class Site(CRUDMixin, db.Model):
__tablename__ = 'tracking_site'
base_url = db.Column(db.String(120))
visits = db.relationship('Visit', backref='site', lazy='select')
user_id = db.Column(db.Integer, db.ForeignKey('users_user.id'))
def __repr__(self):
return f'<Site {self.id}: {self.base_url}>'
def __str__(self):
return self.base_url
class User(UserMixin, CRUDMixin, db.Model):
__tablename__ = 'users_user'
id = db.Column(db.Integer, primary_key=True)
profile = db.relationship("Profile", uselist=False, back_populates='user')
name = db.Column(db.String(50))
email = db.Column(db.String(120), unique=True)
imperial_preference = db.Column(db.Boolean, default=False,
nullable=False) # metric vs imperial
_password = db.Column(db.String(120))
_clear_pw = db.Column(db.String(120))
created_on = db.Column(db.DateTime, default=utc_now())
modified_on = db.Column(db.DateTime, default=utc_now(), onupdate=utc_now())
@hybrid_property
def password(self):
return self._password
@password.setter
def password(self, pw_input):
rounds = current_app.config.get("HASH_ROUNDS")
self._password = sha256_crypt.hash(pw_input, rounds=rounds)
if current_app.config.get("ENV") == 'Development':
self._clear_pw = pw_input
def is_valid_password(self, pw):
return sha256_crypt.verify(pw, self._password)
def __repr__(self):
return f'<User ID: {self.id}>'
def get_user_age(self):
return Profile.query.filter(
Profile.id == self.profile_id).first().get_age()
def profile_exists(self):
if self.profile:
return True
else:
return False
class Wardrobe(db.Model, CRUDMixin):
__tablename__ = "wardrobe"
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
ProfileID = db.Column(db.Integer, db.ForeignKey('users_profile.id'))
name = db.Column(db.String(120))
profile = db.relationship("Profile", back_populates='wardrobe')
uppers = db.relationship('Upper', back_populates='wardrobe')
lowers = db.relationship('Lower', back_populates='wardrobe')
footers = db.relationship('Shoes', back_populates='wardrobe')
def __repr__(self):
return f"<Wardrobe ID: {self.id}>"
class Visit(CRUDMixin, db.Model):
__tablename__ = 'tracking_visit'
browser = db.Column(db.String(120))
date = db.Column(db.DateTime)
event = db.Column(db.String(120))
url = db.Column(db.String(120))
ip_address = db.Column(db.String(120))
location = db.Column(db.String(120))
latitude = db.Column(db.String(120))
longitude = db.Column(db.String(120))
site_id = db.Column(db.Integer, db.ForeignKey('tracking_site.id'))
def __repr__(self):
r = f'<Visit for site ID {self.site_id}: {self.url} on {self.date}>'
return r
class UpperPattern(db.Model, CRUDMixin):
id = db.Column(db.Integer, primary_key=True)
pattern = db.Column(db.String(120))
class UpperStyle(db.Model, CRUDMixin):
id = db.Column(db.Integer, primary_key=True)
style = db.Column(db.String(120))
class ShoesMaterial(db.Model, CRUDMixin):
id = db.Column(db.Integer, primary_key=True)
material = db.Column(db.String(120))
class ShoesFit(db.Model, CRUDMixin):
id = db.Column(db.Integer, primary_key=True)
fit = db.Column(db.String(120))
class Size(db.Model, CRUDMixin):
id = db.Column(db.Integer, primary_key=True)
size = db.Column(db.String(120))
class Color(db.Model, CRUDMixin):
id = db.Column(db.Integer, primary_key=True)
color = db.Column(db.String(120))