37 lines
827 B
Python
37 lines
827 B
Python
from flask.ext.script import Manager, prompt_bool
|
|
from api import app, db, esales
|
|
from api.models import Designer
|
|
|
|
manager = Manager(app)
|
|
|
|
|
|
@manager.command
|
|
def init_db():
|
|
db.create_all()
|
|
print("All tables created")
|
|
|
|
|
|
@manager.command
|
|
def drop_db():
|
|
if prompt_bool("Are you sure you want to lose all your data?"):
|
|
db.drop_all()
|
|
print("All tables dropped")
|
|
|
|
|
|
@manager.command
|
|
def seed_db():
|
|
print("Adding sample data")
|
|
designer = Designer(name="Martin", path='/martin')
|
|
db.session.add(designer)
|
|
db.session.commit()
|
|
|
|
|
|
@manager.command
|
|
def run_esales_import():
|
|
print("Importing products and categories into eSales")
|
|
pattern = "{}/*.xml".format(app.config.get("ESALES_IMPORT_DIR"))
|
|
esales.import_products_and_categories(pattern)
|
|
|
|
if __name__ == '__main__':
|
|
manager.run()
|