123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293 |
- 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)
|