import datetime import peewee import lc.config import lc.requests as r class Model(peewee.Model): class Meta: database = lc.config.DB # TODO: figure out authorization for users (oauth? passwd?) class User(Model): ''' A user! you know tf this is about ''' name = peewee.TextField() class Link(Model): ''' A link as stored in the database ''' url = peewee.TextField() name = peewee.TextField() description = peewee.TextField() # TODO: do we need to track modified time? created = peewee.DateTimeField() # is the field entirely private? private = peewee.BooleanField() # owned by user = peewee.ForeignKeyField(User, backref='all_links') @staticmethod def from_request(user: User, link: r.Link) -> 'Link': l = Link.create( url=link.url, name=link.name, description=link.description, private=link.private, created=datetime.datetime.now(), user=user, ) for tag_name in link.tags: t = Tag.find_tag(tag_name) HasTag.create( link=l, tag=t, ) return l class Tag(Model): ''' A tag. This just indicates that a user has used this tag at some point. ''' name = peewee.TextField() parent = peewee.ForeignKeyField('self', null=True, backref='children') @staticmethod def find_tag(tag_name: str): if (t := Tag.get_or_none(Tag.name == tag_name)): return t parent = None if '/' in tag_name: parent_name = tag_name[:tag_name.rindex('/')] parent = Tag.find_tag(parent_name) return Tag.create( name=tag_name, parent=parent, ) class HasTag(Model): ''' Establishes that a link is tagged with a given tag. ''' link = peewee.ForeignKeyField(Link, backref='tags') tag = peewee.ForeignKeyField(Tag, backref='models') MODELS = [ User, Link, Tag, HasTag, ] def create_tables(): lc.config.DB.create_tables(MODELS, safe=True)