예제 #1
0
import time

from sqlalchemy.orm import sessionmaker

import db
import util

Session = sessionmaker(bind=db.gen_engine(db.get_database_url()))


def set_notas(user, notas_resumo, notas_detalhe, bot):
    if int(time.strftime("%m", time.localtime())) >= 7:
        semestre = str(time.strftime("%Y/2", time.localtime()))
    else:
        semestre = str(time.strftime("%Y/1", time.localtime()))
    for resumo in notas_resumo:
        session = Session()
        notas_resumo_db = session.query(db.NotasResumo).filter_by(
            user_id=user.telegram_id,
            materia=str(resumo[0]),
            semestre=semestre).first()

        user_id = user.telegram_id
        materia = str(resumo[0])
        primeira_av = float(util.verifica_vazio(resumo[1]))
        segunda_av = float(util.verifica_vazio(resumo[2]))
        media = float(util.verifica_vazio(resumo[3]))
        av_complementar = float(util.verifica_vazio(resumo[4]))
        media_final = float(util.verifica_vazio(resumo[5]))

        if not notas_resumo_db:
예제 #2
0
config = ConfigParser()

if not config.read('config.ini'):
    print('Config file not found, generating')
    GenerateConfig()
    config.read('config.ini')

for key in config['db']:
    print(key, config['db'][key])

dbsettings = dict(config.items('db'))
token = config['telegram']['token']

url = getDatabaseUrl(dbsettings)

engine = gen_engine(url)
Session = sessionmaker(bind=engine)

bot = telegram.Bot(token=token)

print(bot.get_me())

updater = Updater(token=token)
dispatcher = updater.dispatcher

logging.basicConfig(
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    level=logging.INFO)


def start(bot, update):