def send_welcome(message):
user_id = message.from_user.id
try:
if 'authenticated' in message.text:
redis_helpers.store_api_user(user_id)
redis_helpers.store_resource('account', user_id)
redis_helpers.store_resource('category', user_id)
BOT.send_message(user_id,
'Authentication successful!',
reply_markup=DEFAULT_KEYBOARD)
else:
welcome_msg = ('''
Hi, %(username)s!\n\nI\'m simple helper bot for [MoneyKeeper](%(url)s).
My responsibilities are: send your transactions to [MoneyKeeper](%(url)s) and show info from your accounts.
To proceed you will need authenticate [here](%(auth_url)s)
In case of mistrust you can visit the source code on my creators [GitHub](https://github.com/ilov3).
''')
kwargs = {
'username': message.from_user.username or 'stranger',
'url': MONEYKEEPER_URL,
'auth_url': MONEYKEEPER_URL + 'auth/%s' % user_id
}
BOT.send_message(user_id,
welcome_msg % kwargs,
reply_markup=DEFAULT_KEYBOARD,
parse_mode='Markdown',
disable_web_page_preview=True)
except Exception as e:
logger.error(e)
def account_chosen(message):
BOT.register_next_step_handler(message, amount_given)
account = message.text
user_id = message.from_user.id
redis_helpers.store_transaction_from(account, user_id)
BOT.send_message(user_id,
'Give amount:',
reply_markup=types.ReplyKeyboardHide())
def account_to_chosen(message):
BOT.register_next_step_handler(message, account_chosen)
account_to = message.text
user_id = message.from_user.id
redis_helpers.store_transaction_to(account_to, user_id)
accounts = redis_helpers.get_stored_resource('account', user_id)
markup = types.ReplyKeyboardMarkup(one_time_keyboard=True,
resize_keyboard=True)
markup.add(*[acc['name'].encode('utf-8') for acc in accounts])
markup.add('/cancel')
BOT.send_message(user_id, 'Choose account:', reply_markup=markup)
def transfer(message):
BOT.register_next_step_handler(message, account_to_chosen)
user_id = message.from_user.id
redis_helpers.store_transaction_kind('transfer', user_id)
try:
accounts = redis_helpers.get_stored_resource('account', user_id)
markup = types.ReplyKeyboardMarkup(one_time_keyboard=True, resize_keyboard=True)
markup.add(*[acc['name'].encode('utf-8') for acc in accounts])
markup.add('/cancel')
BOT.send_message(user_id, 'Choose account transfer to:', reply_markup=markup)
except Exception as e:
logger.warn(e)
def send_help(message):
transaction_kinds = ['income', 'expense', 'transfer']
user_id = message.from_user.id
help_msg = ('''
/help - shows this help message
/start - intro message
/info - shows accounts balances
/update - update your accounts and categories
/cancel - cancels transaction preparation
%s
''')
BOT.send_message(user_id, help_msg % '\n'.join(['/{0} - post {0} transaction'.format(kind) for kind in transaction_kinds]))
def expense(message):
BOT.register_next_step_handler(message, category_chosen)
user_id = message.from_user.id
redis_helpers.store_transaction_kind('expense', user_id)
try:
categories = redis_helpers.get_stored_resource('category', user_id)
markup = types.ReplyKeyboardMarkup(one_time_keyboard=True, resize_keyboard=True)
markup.add(*[cat['name'].encode('utf-8') for cat in categories if cat['kind'] == 'exp'])
markup.add('/cancel')
BOT.send_message(user_id, 'Choose category:', reply_markup=markup)
except Exception as e:
logger.warn(e)
def amount_given(message):
user_id = message.from_user.id
try:
amount = message.text
redis_helpers.store_transaction_amount(amount, user_id)
code = send_transaction(user_id)
logger.info(code)
if code == 201:
get_info(user_id)
else:
BOT.send_message(user_id, 'Something went wrong...')
except Exception as e:
logger.error(e)
def amount_given(message):
user_id = message.from_user.id
try:
amount = message.text
redis_helpers.store_transaction_amount(amount, user_id)
code = send_transaction(user_id)
logger.info(code)
if code == 201:
get_info(user_id)
else:
BOT.send_message(user_id, 'Something went wrong...')
except Exception as e:
logger.error(e)
def transfer(message):
BOT.register_next_step_handler(message, account_to_chosen)
user_id = message.from_user.id
redis_helpers.store_transaction_kind('transfer', user_id)
try:
accounts = redis_helpers.get_stored_resource('account', user_id)
markup = types.ReplyKeyboardMarkup(one_time_keyboard=True,
resize_keyboard=True)
markup.add(*[acc['name'].encode('utf-8') for acc in accounts])
markup.add('/cancel')
BOT.send_message(user_id,
'Choose account transfer to:',
reply_markup=markup)
except Exception as e:
logger.warn(e)
def expense(message):
BOT.register_next_step_handler(message, category_chosen)
user_id = message.from_user.id
redis_helpers.store_transaction_kind('expense', user_id)
try:
categories = redis_helpers.get_stored_resource('category', user_id)
markup = types.ReplyKeyboardMarkup(one_time_keyboard=True,
resize_keyboard=True)
markup.add(*[
cat['name'].encode('utf-8') for cat in categories
if cat['kind'] == 'exp'
])
markup.add('/cancel')
BOT.send_message(user_id, 'Choose category:', reply_markup=markup)
except Exception as e:
logger.warn(e)
def send_help(message):
transaction_kinds = ['income', 'expense', 'transfer']
user_id = message.from_user.id
help_msg = ('''
/help - shows this help message
/start - intro message
/info - shows accounts balances
/update - update your accounts and categories
/cancel - cancels transaction preparation
%s
''')
BOT.send_message(
user_id, help_msg % '\n'.join([
'/{0} - post {0} transaction'.format(kind)
for kind in transaction_kinds
]))
def send_welcome(message):
user_id = message.from_user.id
try:
if 'authenticated' in message.text:
redis_helpers.store_api_user(user_id)
redis_helpers.store_resource('account', user_id)
redis_helpers.store_resource('category', user_id)
BOT.send_message(user_id, 'Authentication successful!', reply_markup=DEFAULT_KEYBOARD)
else:
welcome_msg = ('''
Hi, %(username)s!\n\nI\'m simple helper bot for [MoneyKeeper](%(url)s).
My responsibilities are: send your transactions to [MoneyKeeper](%(url)s) and show info from your accounts.
To proceed you will need authenticate [here](%(auth_url)s)
In case of mistrust you can visit the source code on my creators [GitHub](https://github.com/ilov3).
''')
kwargs = {
'username': message.from_user.username or 'stranger',
'url': MONEYKEEPER_URL,
'auth_url': MONEYKEEPER_URL + 'auth/%s' % user_id
}
BOT.send_message(user_id, welcome_msg % kwargs, reply_markup=DEFAULT_KEYBOARD, parse_mode='Markdown', disable_web_page_preview=True)
except Exception as e:
logger.error(e)
def update(message):
user_id = message.from_user.id
redis_helpers.update_stored_resource('account', user_id)
redis_helpers.update_stored_resource('category', user_id)
BOT.send_message(user_id, 'Data updated!', reply_markup=DEFAULT_KEYBOARD)
def update(message):
user_id = message.from_user.id
redis_helpers.update_stored_resource('account', user_id)
redis_helpers.update_stored_resource('category', user_id)
BOT.send_message(user_id, 'Data updated!', reply_markup=DEFAULT_KEYBOARD)
def cancel(message):
user_id = message.from_user.id
redis_helpers.flush_transaction_fields(user_id)
BOT.send_message(user_id, 'Transaction cancelled!', reply_markup=DEFAULT_KEYBOARD)
BOT.send_message(user_id, 'Choose account transfer to:', reply_markup=markup)
except Exception as e:
logger.warn(e)
@BOT.message_handler(commands=['info'])
@login_required
def info(message):
user_id = message.from_user.id
redis_helpers.update_stored_resource('account', user_id)
get_info(user_id)
@BOT.message_handler(commands=['cancel'])
def cancel(message):
user_id = message.from_user.id
redis_helpers.flush_transaction_fields(user_id)
BOT.send_message(user_id, 'Transaction cancelled!', reply_markup=DEFAULT_KEYBOARD)
@BOT.message_handler(commands=['update'])
@login_required
def update(message):
user_id = message.from_user.id
redis_helpers.update_stored_resource('account', user_id)
redis_helpers.update_stored_resource('category', user_id)
BOT.send_message(user_id, 'Data updated!', reply_markup=DEFAULT_KEYBOARD)
BOT.polling(none_stop=True, interval=0, timeout=3)
def account_chosen(message):
BOT.register_next_step_handler(message, amount_given)
account = message.text
user_id = message.from_user.id
redis_helpers.store_transaction_from(account, user_id)
BOT.send_message(user_id, 'Give amount:', reply_markup=types.ReplyKeyboardHide())
def get_info(user_id):
accounts = get_stored_resource('account', user_id)
balances = '\n'.join(['%s: %s' % (acc['name'], acc['get_balance']) for acc in accounts])
balances += '\nTotal: %s' % sum([acc['get_balance'] for acc in accounts])
BOT.send_message(user_id, balances, reply_markup=DEFAULT_KEYBOARD)
def cancel(message):
user_id = message.from_user.id
redis_helpers.flush_transaction_fields(user_id)
BOT.send_message(user_id,
'Transaction cancelled!',
reply_markup=DEFAULT_KEYBOARD)
logger.warn(e)
@BOT.message_handler(commands=['info'])
@login_required
def info(message):
user_id = message.from_user.id
redis_helpers.update_stored_resource('account', user_id)
get_info(user_id)
@BOT.message_handler(commands=['cancel'])
def cancel(message):
user_id = message.from_user.id
redis_helpers.flush_transaction_fields(user_id)
BOT.send_message(user_id,
'Transaction cancelled!',
reply_markup=DEFAULT_KEYBOARD)
@BOT.message_handler(commands=['update'])
@login_required
def update(message):
user_id = message.from_user.id
redis_helpers.update_stored_resource('account', user_id)
redis_helpers.update_stored_resource('category', user_id)
BOT.send_message(user_id, 'Data updated!', reply_markup=DEFAULT_KEYBOARD)
BOT.polling(none_stop=True, interval=0, timeout=3)
def wrapper(message):
user_id = message.from_user.id
if get_api_token(user_id):
func(message)
else:
BOT.send_message(user_id, 'Authenticate first, through /start command.')
def account_to_chosen(message):
BOT.register_next_step_handler(message, account_chosen)
account_to = message.text
user_id = message.from_user.id
redis_helpers.store_transaction_to(account_to, user_id)
accounts = redis_helpers.get_stored_resource('account', user_id)
markup = types.ReplyKeyboardMarkup(one_time_keyboard=True, resize_keyboard=True)
markup.add(*[acc['name'].encode('utf-8') for acc in accounts])
markup.add('/cancel')
BOT.send_message(user_id, 'Choose account:', reply_markup=markup)