def start(bot, update, chat_data, job_queue):
if 'is_notifications' not in chat_data:
chat_data['is_notifications'] = True
job = job_queue.run_daily(birthdays.alarm,
time(9, 0),
days=(0, 1, 2, 3, 4, 5, 6),
context=[update, chat_data],
name='default')
chat_data['job'] = job
if 'time_zone' not in chat_data:
chat_data['time_zone'] = time(0, 0)
if 'sign' not in chat_data:
chat_data['sign'] = True
if 'user_time' not in chat_data:
chat_data['user_time'] = time(9, 0)
messages = DBMessages.query.filter(
DBMessages.user == update.message.chat_id).all()
if len(messages) == 0:
user = DBMessages(update.message.from_user.id)
print(update.message.from_user.username, update.message.from_user.id)
db_session.add(user)
db_session.commit()
key_board = [["/add"], ["/month", "/help"]]
bot.sendMessage(
update.message.chat_id,
text="Hi! I'm <b>BirthdayBot!</b>\n"
"Press /help for more information\n\n"
"If you're new here I highly recommend to start with settings "
"by using /set command\n\n",
parse_mode='HTML',
reply_markup=ReplyKeyboardMarkup(key_board, one_time_keyboard=False))
return 'Menu'
def mutate(root, info, area_data=None):
try:
area = AreaModel(**area_data)
db_session.add(area)
db_session.commit()
except BaseException as err:
db_session.rollback()
raise GraphQLError("Error inserting Area... {}".format(err))
def mutate(root, info, catalog_data=None):
try:
catalog = CatalogModel(**catalog_data)
db_session.add(catalog)
db_session.commit()
except BaseException as err:
db_session.rollback()
raise GraphQLError("Error inserting Catalog... {}".format(err))
def mutate(root, info, item_data=None):
try:
item = ItemModel(**item_data)
db_session.add(item)
db_session.commit()
except BaseException as err:
db_session.rollback()
raise GraphQLError("Error inserting Item... {}".format(err))
def mutate(root, info, task_data=None):
try:
task = TaskModel(**task_data)
db_session.add(task)
db_session.commit()
return CreateTask(task=task, status=True)
except BaseException as err:
db_session.rollback()
raise GraphQLError("Error inserting Task... {}".format(err))
def mutate(self, info, name, author, price, title=None):
create_time = update_time = time.time()
id = new_id()
book = models.ExchangeAccount(id=id, name=name, title=title,
price=price, author=author, del_flag=1, create_time=create_time,
update_time=update_time)
db_session.add(book)
db_session.commit()
return addBook(book=book)
def login():
"""Logs the user in or creates a new user in the database."""
form = LoginForm(request.form)
validates = request.method == 'POST' and form.validate()
if validates:
username = request.form.get('username')
password = request.form.get('password')
if not User.username_taken(username):
user = User(username=username, password=hashulate(password))
db_session.add(user)
db_session.commit()
login_user(user)
return redirect(request.args.get("next") or url_for("home"))
elif User.check_password(User.id_from_name(username),
request.form.get('password')):
login_user(User(id=User.id_from_name(username)))
return redirect(request.args.get("next") or url_for("home"))
return render_template('login.html', form=form)
def load_pickled_data():
with open('cocktails.pkl', 'rb') as f:
pickled_cocktails = pickle.load(f)
for c in pickled_cocktails:
# instantiate and add cocktail
cocktail = Cocktail(name=c.name, glass=c.glass, recipe=c.recipe, image=c.image)
db_session.add(cocktail)
# add each ingredient that is not already in db
for i in c.ingredients:
ingredient = Ingredient.query.filter(Ingredient.name==i[0]).one_or_none()
if ingredient is None:
image_name = i[0].replace(' ', '+').replace('/', '\\')
image_path = '{0}/{1}'.format('ingredients', image_name)
ingredient = Ingredient(i[0], image_path)
db_session.add(ingredient)
# add the amount for each ingredient
amount = Amount(cocktail, ingredient, i[1])
db_session.add(amount)
# commit unpickled cocktail goodness
db_session.commit()
def process(): recs = db_session.query(Eq).filter_by().all() for record in recs: wpname = record.WeaponName wptype = record.WeaponType arname = record.ArmorName artype = record.ArmorType charid = record.CHARACTERS_CharacterID print wpname,wptype,arname,artype,charid if wpname != None: rec = db_session.query(Equipment).filter_by(Name = wpname, Type = wptype, EquipmentType = 'Weapon').first() else: rec = db_session.query(Equipment).filter_by(Name = arname, Type = artype, EquipmentType = 'Armor').first() obj = CharHasEquip(record.CHARACTERS_CharacterID,rec.RowID) db_session.add(obj) db_session.commit()
def add_month_add_birthday(bot, update):
messages = DBMessages.query.filter(
DBMessages.user == update.message.from_user.id).first()
day = messages.day
month = update.message.text.strip()
name = messages.name
link = messages.link
if month not in birthday_class.MonthToNumber:
reply_keyboard = [['January', 'February', 'March'],
['April', 'May', 'June'],
['July', 'August', 'September'],
['October', 'November', 'December']]
update.message.reply_text("Wrong month \n",
reply_markup=ReplyKeyboardMarkup(
reply_keyboard, one_time_keyboard=True))
return "Add_AddMonthAddBirthday"
if month == "February" and (str(day) == '31' or str(day) == '30'):
key_board = [["/add"], ["/month", "/year"]]
update.message.reply_text(text="No such day in February:(",
reply_markup=ReplyKeyboardMarkup(
key_board, one_time_keyboard=False))
return "Menu"
if month in {'April', 'June', 'September', 'November'
} and str(day) == '31':
key_board = [["/add"], ["/month", "/year"]]
update.message.reply_text(text="This month contains only 30 days;)",
reply_markup=ReplyKeyboardMarkup(
key_board, one_time_keyboard=False))
return "Menu"
n_date = date(2016, birthday_class.MonthToNumber[month], int(day))
new_birthday = DBBirthdays(update.message.chat_id, n_date, name, link)
db_session.add(new_birthday)
db_session.commit()
key_board = [["/add"], ["/month", "/year"]]
update.message.reply_text(text="New birthday successfully added",
reply_markup=ReplyKeyboardMarkup(
key_board, one_time_keyboard=False))
return "Menu"
''' Created on 03/03/2013 @author: Administrador ''' from app import app from app.modelo.autenticacion.Usuario import Usuario from config import init_db, db_session init_db() u=Usuario(); u.apellido='a' u.email='*****@*****.**' u.nombre='aaa' u.telefono='aaa' u.user='******' db_session.add(u) app.run(debug=False)
