FitTrackee/fittrackee_api/server.py

95 lines
2.7 KiB
Python
Raw Normal View History

import shutil
from fittrackee_api import create_app, db
from fittrackee_api.activities.models import Activity, Sport
from fittrackee_api.activities.utils import update_activity
from fittrackee_api.application.utils import init_config
from fittrackee_api.users.models import User
from tqdm import tqdm
2017-12-16 21:00:46 +01:00
2018-04-09 22:09:58 +02:00
app = create_app()
2017-12-16 21:00:46 +01:00
@app.cli.command()
2018-01-20 19:12:34 +01:00
def drop_db():
2018-05-10 11:36:08 +02:00
"""Empty database for dev environments."""
db.engine.execute("DROP TABLE IF EXISTS alembic_version;")
2017-12-16 21:00:46 +01:00
db.drop_all()
2018-01-20 19:12:34 +01:00
db.session.commit()
print('Database dropped.')
shutil.rmtree(app.config['UPLOAD_FOLDER'], ignore_errors=True)
print('Uploaded files deleted.')
2018-01-20 19:12:34 +01:00
@app.cli.command()
2018-11-18 13:55:21 +01:00
def initdata():
2018-01-20 19:12:34 +01:00
"""Init the database."""
2018-01-01 16:59:46 +01:00
admin = User(
2019-08-28 13:25:39 +02:00
username='admin', email='admin@example.com', password='mpwoadmin'
)
2017-12-17 09:16:08 +01:00
admin.admin = True
admin.timezone = 'Europe/Paris'
2017-12-17 09:16:08 +01:00
db.session.add(admin)
2018-05-16 23:52:55 +02:00
sport = Sport(label='Cycling (Sport)')
sport.img = '/img/sports/cycling-sport.png'
sport.is_default = True
db.session.add(sport)
sport = Sport(label='Cycling (Transport)')
sport.img = '/img/sports/cycling-transport.png'
sport.is_default = True
db.session.add(sport)
sport = Sport(label='Hiking')
sport.img = '/img/sports/hiking.png'
sport.is_default = True
db.session.add(sport)
sport = Sport(label='Mountain Biking')
sport.img = '/img/sports/mountain-biking.png'
sport.is_default = True
db.session.add(sport)
sport = Sport(label='Running')
sport.img = '/img/sports/running.png'
sport.is_default = True
db.session.add(sport)
sport = Sport(label='Walking')
sport.img = '/img/sports/walking.png'
sport.is_default = True
db.session.add(sport)
2017-12-16 21:00:46 +01:00
db.session.commit()
print('Initial data stored in database.')
2017-12-16 21:00:46 +01:00
@app.cli.command()
def recalculate():
print("Starting activities data refresh")
2019-08-28 13:25:39 +02:00
activities = (
Activity.query.filter(Activity.gpx != None)
.order_by(Activity.activity_date.asc()) # noqa
.all()
)
if len(activities) == 0:
print('➡️ no activities to upgrade.')
return None
pbar = tqdm(activities)
for activity in pbar:
update_activity(activity)
pbar.set_postfix(activitiy_id=activity.id)
db.session.commit()
@app.cli.command('init-app-config')
def init_app_config():
"""Init application configuration."""
print("Init application configuration")
config_created, _ = init_config()
if config_created:
print("Creation done!")
else:
print(
"Application configuration already existing in database. "
"Please use web application to update it."
)
2017-12-16 21:00:46 +01:00
if __name__ == '__main__':
app.run()