101 lines
2.3 KiB
Python
101 lines
2.3 KiB
Python
# region ############################# IMPORTS #############################
|
|
|
|
import logging
|
|
from debug import setup_logging
|
|
log = logging.getLogger("default")
|
|
setup_logging()
|
|
|
|
import os
|
|
from datetime import datetime
|
|
from peewee import *
|
|
from playhouse.sqlite_ext import SqliteExtDatabase#, FTS5Model, SearchField
|
|
from configuration import read_config, write_config
|
|
from paths import DATA_DIR
|
|
|
|
# endregion
|
|
|
|
|
|
# region ############################# GLOBALS #############################
|
|
realpath = os.path.dirname(os.path.realpath(__file__))
|
|
rp = realpath
|
|
|
|
db_path = os.path.join(DATA_DIR, 'database.db')
|
|
pragmas = [
|
|
('journal_mode', 'wal'),
|
|
('cache_size', -1000 * 32)]
|
|
db = SqliteExtDatabase(db_path, pragmas=pragmas)
|
|
|
|
# endregion
|
|
|
|
|
|
|
|
# region ############################# TABLE CLASSES #############################
|
|
|
|
class BroModel(Model):
|
|
date_created = DateTimeField(default=datetime.now())
|
|
date_updated = DateTimeField(default=datetime.now())
|
|
date_deleted = DateTimeField(null=True)
|
|
deleted = BooleanField(default=False)
|
|
|
|
def mark_deleted(self):
|
|
self.deleted = True
|
|
self.date_deleted = datetime.now()
|
|
self.save()
|
|
|
|
class Entry(BroModel):
|
|
upload = FloatField(null=True)
|
|
download = FloatField(null=True)
|
|
data = TextField(null=True)
|
|
|
|
|
|
class Meta:
|
|
database = db
|
|
|
|
def create(self, **query):
|
|
ret = super(Entry, self).create(**query)
|
|
return ret
|
|
|
|
def save(self, *args, **kwargs):
|
|
self.date_updated = datetime.now()
|
|
ret = super(Entry, self).save(*args, **kwargs)
|
|
|
|
return ret
|
|
|
|
|
|
|
|
# region Migration
|
|
config = read_config()
|
|
if config['database_migrate']:
|
|
log.debug("=====================")
|
|
log.debug("Migration stuff...")
|
|
try:
|
|
from playhouse.migrate import *
|
|
|
|
migrator = SqliteMigrator(db)
|
|
|
|
data = TextField(null=True)
|
|
|
|
migrate(
|
|
migrator.add_column('Entry', 'data', data)
|
|
)
|
|
log.debug("Migration success")
|
|
log.debug("=====================")
|
|
|
|
config['database_migrate'] = False
|
|
write_config(config)
|
|
except:
|
|
log.error("Could not migrate", exc_info=True)
|
|
log.debug("=====================")
|
|
else:
|
|
db.connect()
|
|
db.create_tables([Entry])
|
|
# endregion
|
|
|
|
log.info(" ".join(["Using DB", str(db), "At path:", str(db_path)]))
|
|
|
|
# On init make sure we create database
|
|
|
|
|
|
|
|
# endregion
|