hamster-tools/hamstertools/db.py

62 lines
1.2 KiB
Python

import logging
from peewee import SqliteDatabase, Model, CharField, ForeignKeyField
from textual.logging import TextualHandler
logger = logging.getLogger('peewee')
logger.addHandler(TextualHandler())
logger.setLevel(logging.DEBUG)
db = SqliteDatabase(None)
class HamsterCategory(Model):
name = CharField()
class Meta:
database = db
table_name = 'categories'
class HamsterActivity(Model):
name = CharField()
category = ForeignKeyField(HamsterCategory, backref='activities')
class Meta:
database = db
table_name = 'activities'
class HamsterFact(Model):
activity = ForeignKeyField(HamsterActivity, backref='facts')
class Meta:
database = db
table_name = 'facts'
class KimaiCustomer(Model):
name = CharField()
class Meta:
database = db
table_name = 'kimai_customers'
class KimaiProject(Model):
name = CharField()
customer = ForeignKeyField(KimaiCustomer, backref='projects')
class Meta:
database = db
table_name = 'kimai_projects'
class KimaiActivity(Model):
name = CharField()
project = ForeignKeyField(KimaiProject, backref='activities')
class Meta:
database = db
table_name = 'kimai_activities'