Source code for frostmark.db
'''
Module for creating SQLite DB schema and session retrieving.
'''
from frostmark.db_base import BASE, SESSIONMAKER, ENGINE
[docs]def folder_check_root(session):
'''
Check whether default rows for Folder model exist and create them if not.
'''
from frostmark.models import Folder
folders = session.query(Folder).all()
root_folder = Folder.get_root()
if not any([fold.id == root_folder[0] for fold in folders]):
session.add(Folder(
id=root_folder[0],
folder_name=root_folder[1],
parent_folder_id=root_folder[0]
))
session.commit()
[docs]def get_session():
'''
Create DB schema and return a new session.
'''
BASE.metadata.create_all(ENGINE)
session = SESSIONMAKER()
folder_check_root(session)
return session