def create_db():
from app.database import create_tables
create_tables()
def reset_db():
from app.database import create_tables, drop_tables
drop_tables()
create_tables()
def create_tables():
# Create table for each model if it does not exist.
database.create_tables([Entry, Tag, EntryTags, FTSEntry], safe=True)
@login_required
def delete_feed():
feed = request.form['feed']
try:
fd = Feed.get(Feed.id == feed, Feed.user == current_user.id)
fd.delete_instance()
return jsonify(status='OK')
except Feed.DoesNotExist:
return jsonify(status='FAIL')
@app.route("/rss/<feed>")
@login_required
def rss(feed):
try:
fd = Feed.get(Feed.id == feed, Feed.user == current_user.id)
return render_template("rss.html",
feed=fd,
entries=feedparser.parse(fd.url).entries)
except Feed.DoesNotExist:
abort(404)
@app.errorhandler(404)
def not_found(exc):
return render_template("notfound.html"), 404
if __name__ == "__main__":
database.create_tables([User, Feed], safe=True)
app.run()
def create_tables():
database.connect()
database.create_tables([Host, CPUStats], safe=False)
def main():
database.create_tables([User, Feed], safe=True)
app.run()
def initialize_models():
database.connect()
database.create_tables([PlainTextUpload], True)
database.close()
def main():
database.create_tables([Entry, FTSEntry], safe=True)
app.run(debug=True)
def setUp(self):
self.app = create_app('testing')
self.test_app = self.app.test_client()
create_tables()
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
from config import PATH_SQLITE_DB
from app import database, Entry, FTSEntry
# init database if don't exist
if not os.path.exists(PATH_SQLITE_DB):
database.create_tables([Entry, FTSEntry], safe=True)
def droptable(module, name):
model = importlib.import_module(f"common.models.{module}")
cls = getattr(model, name)
database.drop_tables([cls], safe=True)
database.create_tables([cls], safe=False)
def dbinit():
createModels = []
for m in app.config.get('INSTALL_MODELS', []):
modules = importlib.import_module(f"common.models.{m}")
createModels += getAllmodel(modules)
database.create_tables(createModels, safe=True)
def create_tables():
database.connect()
database.create_tables([Host, CPUStats], safe=False)
def new_database(postgresql):
"""
Resets the db inside the module to a new mocked one
"""
database.db = postgresql
database.create_tables()
def configureDB():
database.create_tables([Entry, FTSEntry], safe=True)
