manage.py

"""
UniveCalendar manager

Usage Example
-------------
> python manage.py runserver

"""
import os
import sys

from flask.ext.migrate import Migrate, MigrateCommand
from flask.ext.script import Manager

from app import create_app, db, models, bot

app = create_app(os.getenv('FLASK_CONFIG') or 'development')

migrate = Migrate(app, db)
manager = Manager(app)

manager.add_command('db', MigrateCommand)


@bot.message_handler(commands=['start', 'help'])
def connect_user(msg):
    token = msg.text.split()[1] if len(msg.text.split()) > 1 else None
    if token:
        with app.app_context():
            u = models.User.load_user_from_unique_code(token)
            if u:
                u.telegram_chat_id = msg.chat.id
                db.session.commit()
                reply = "Congratulations {0}, you successfully connect your account.".format(u.username)
            else:
                reply = "I have no clue who you are..."
    else:
        reply = """
        Hi! I'm UniveCalendar Bot. I can notify you about events of
        your university's calendar. Visit our page for more info.

        Available Commands
        /start <TOKEN> - Authorize bot
        /stop - Disconnect your connect

        """
    bot.reply_to(msg, reply)


@bot.message_handler(commands=['stop'])
def stop(msg):
    with app.app_context():
        u = models.User.query.filter(models.User.telegram_chat_id == str(msg.chat.id)).first()
        if u:
            u.telegram_chat_id = None
            db.session.commit()
            reply = "Ok, I will not send you any notifications."
        else:
            reply = "I have no clue who you are..."
    bot.reply_to(msg, reply)


@manager.command
def runbot():
    import logging
    from telebot import logger
    logger.setLevel(logging.INFO)
    bot.polling()


@manager.command
def cleardb():
    """Delete all tables records."""
    for table in reversed(db.metadata.sorted_tables):
        db.session.execute(table.delete())
    db.session.commit()


@manager.command
def cleartable(tablename):
    """Delete all tables records."""
    table = db.metadata.tables[tablename]
    db.session.execute(table.delete())
    db.session.commit()


@manager.command
def insertfeed(professor_id):
    import forgery_py

    p = models.Professor.query.get(professor_id)
    f = models.Feed(title=forgery_py.lorem_ipsum.title(),
                    body=forgery_py.lorem_ipsum.paragraphs(2),
                    author=p)

    db.session.add(f)
    db.session.commit()


@manager.command
def delaylesson(lesson_id, hours=1):
    from datetime import timedelta

    l = models.Lesson.query.get(lesson_id)
    l.start += timedelta(hours=hours)
    l.end += timedelta(hours=hours)
    db.session.commit()


@manager.command
def addusers():
    u = models.User(email='bob@unive.it',
                    username='bobby85',
                    confirmed=True,
                    password='123456')
    db.session.add(u)
    db.session.commit()

    u = models.User(email='alice@unive.it',
                username='alice92',
                confirmed=True,
                password='123456')
    db.session.add(u)
    db.session.commit()


@manager.command
def dbcreate():
    os.system('python manage.py db init')
    os.system('python manage.py db migrate')
    os.system('python manage.py db upgrade')


@manager.command
def trcompile():
    if sys.platform == 'win32':
        pybabel = 'flask\\Scripts\\pybabel'
    else:
        pybabel = 'pybabel'
    os.system(pybabel + ' compile -d app/translations')


@manager.command
def trupdate():
    if sys.platform == 'win32':
        pybabel = 'flask\\Scripts\\pybabel'
    else:
        pybabel = 'pybabel'
    os.system(pybabel + ' extract -F babel.cfg -k lazy_gettext -o messages.pot app')
    os.system(pybabel + ' update -i messages.pot -d app/translations')
    os.unlink('messages.pot')


@manager.command
def trinit():
    if sys.platform == 'win32':
        pybabel = 'flask\\Scripts\\pybabel'
    else:
        pybabel = 'pybabel'
    if len(sys.argv) != 2:
        print "usage: tr_init <language-code>"
        sys.exit(1)
    os.system(pybabel + ' extract -F babel.cfg -k lazy_gettext -o messages.pot app')
    os.system(pybabel + ' init -i messages.pot -d app/translations -l ' + sys.argv[1])
    os.unlink('messages.pot')


if __name__ == '__main__':
    manager.run()