def test_user_exist(self): u1 = models.User('just-an-dev') u2 = models.User('Just-An-dEv') self.assertEqual(True, u1.is_registered()) self.assertEqual(False, u2.is_registered()) self.assertEqual(True, user_function.user_exist('just-an-dev')) self.assertEqual(True, user_function.user_exist('Just-An-dEv'))
def tip_user(rpc, msg): print('An user mention detected ') split_message = msg.body.strip().split() tip_index = split_message.index('+/u/sodogetiptest') if split_message[tip_index] == '+/u/sodogetiptest' and split_message[ tip_index + 2] == 'doge': amount = split_message[tip_index + 1] if utils.check_amount_valid(amount): parent_comment = msg.parent() if user_function.user_exist(msg.author.name): # check we have enough user_balance = crypto.get_user_balance(rpc, msg.author.name) if int(amount) >= user_balance: print( 'user %s not have enough to tip this amount (%s), balance = %s' % (msg.author.name, amount, user_balance)) msg.reply('+/u/%s your balance is too low for this tip ' % msg.author.name) else: # check user have address before tip if user_function.user_exist(parent_comment.author.name): if crypto.tip_user(rpc, msg.author.name, parent_comment.author.name, amount): user_function.add_to_history( msg.author.name, msg.author.name, parent_comment.author.name, amount, "tip") print '%s tip %s to %s' % ( msg.author.name, str(amount), parent_comment.author.name) msg.reply('+/u/%s tip %s to %s' % (msg.author.name, str(amount), parent_comment.author.name)) else: user_function.save_unregistered_tip( msg.author.name, parent_comment.author.name, amount) user_function.add_to_history( msg.author.name, msg.author.name, parent_comment.author.name, amount, "tip", False) print('user %s not registered' % parent_comment.author.name) msg.reply( '+/u/%s need %s before can be tipped (tip saved during 3 day)' % (parent_comment.author.name, linkRegister)) else: msg.reply('You need %s before' % linkRegister) else: print('You must use valid amount') msg.reply('You must use valid amount')
def withdraw_user(rpc, msg): split_message = msg.body.strip().split() if user_function.user_exist(msg.author.name): sender_address = user_function.get_user_address(msg.author.name) amount = split_message[1] user_balance = crypto.get_user_balance(rpc, msg.author.name) if int(amount) >= user_balance: print( 'user %s not have enough to withdraw this amount (%s), balance = %s' % (msg.author.name, amount, user_balance)) msg.reply('+/u/%s your balance is too low for this withdraw ' % msg.author.name) else: if utils.check_amount_valid(amount): receiver_address = split_message[4] try: if crypto.send_to(rpc, sender_address, receiver_address, amount, True): user_function.add_to_history(msg.author.name, sender_address, receiver_address, amount, "withdraw") msg.reply('Withdraw : ' + str(amount) + ' to ' + receiver_address) except: traceback.print_exc() else: print('You must use valid amount') msg.reply('You must use valid amount') else: msg.reply('You need %s before' % linkRegister)
def balance_user(rpc, msg): if user_function.user_exist(msg.author.name): balance = crypto.get_user_confirmed_balance(rpc, msg.author.name) pending_balance = crypto.get_user_unconfirmed_balance( rpc, msg.author.name) spendable_balance = crypto.get_user_spendable_balance( rpc, msg.author.name) + balance bot_logger.logger.info('user %s balance = %s' % (msg.author.name, balance)) balance_value_usd = utils.get_coin_value(balance) pending_value_usd = utils.get_coin_value(pending_balance) spendable_value_usd = utils.get_coin_value(spendable_balance) msg.reply( Template(lang.message_balance + lang.message_footer).render( username=msg.author.name, balance=str(balance), balance_value_usd=str(balance_value_usd), pendingbalance=str(pending_balance), pending_value_usd=str(pending_value_usd), spendablebalance=str(spendable_balance), spendable_value_usd=str(spendable_value_usd))) user_function.add_to_history(msg.author.name, "", "", balance, "balance") else: bot_logger.logger.info('user %s not registered ' % msg.author.name) msg.reply( Template(lang.message_need_register + lang.message_footer).render(username=msg.author.name))
def info_user(rpc, msg): if user_function.user_exist(msg.author.name): address = user_function.get_user_address(msg.author.name) balance = crypto.get_user_confirmed_balance(rpc, msg.author.name) pending_balance = crypto.get_user_unconfirmed_balance( rpc, msg.author.name) spendable_balance = crypto.get_user_spendable_balance( rpc, msg.author.name) + balance balance_value_usd = utils.get_coin_value(balance) pending_value_usd = utils.get_coin_value(pending_balance) spendable_value_usd = utils.get_coin_value(spendable_balance) msg.reply( Template(lang.message_account_details + lang.message_footer).render( username=msg.author.name, balance=str(balance), balance_value_usd=str(balance_value_usd), pendingbalance=str(pending_balance), pending_value_usd=str(pending_value_usd), spendablebalance=str(spendable_balance), spendable_value_usd=str(spendable_value_usd), address=address)) else: msg.reply( Template(lang.message_need_register + lang.message_footer).render(username=msg.author.name))
def register_user(rpc, msg): if not user_function.user_exist(msg.author.name): address = rpc.getnewaddress("reddit-%s" % msg.author.name) if address: msg.reply( Template(lang.message_register_success + lang.message_footer).render(username=msg.author.name, address=address)) user_function.add_user(msg.author.name, address) user_function.add_to_history(msg.author.name, "", "", "", "register") # create a backup of wallet rpc.backupwallet( config.backup_wallet_path + "backup_" + datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S') + ".dat") else: bot_logger.logger.warning('Error during register !') else: bot_logger.logger.info('%s are already registered ' % msg.author.name) balance = crypto.get_user_confirmed_balance(rpc, msg.author.name) address = user_function.get_user_address(msg.author.name) msg.reply( Template(lang.message_already_registered + lang.message_account_details + lang.message_footer).render(username=msg.author.name, address=address, balance=str(balance)))
def balance_user(rpc, msg): if user_function.user_exist(msg.author.name): balance = crypto.get_user_balance(rpc, msg.author.name) print('user %s balance = %s' % (msg.author.name, balance)) value_usd = utils.get_coin_value(balance) msg.reply('Your balance : ' + str(balance) + ' ( ' + str(value_usd) + '$ ) ') else: print('user %s not registered ' % (msg.author.name)) msg.reply('You need %s before' % linkRegister)
def help_user(rpc, msg): if user_function.user_exist(msg.author.name): address = user_function.get_user_address(msg.author.name) msg.reply( Template(lang.message_help + lang.message_footer).render( username=msg.author.name, address=address)) else: msg.reply( Template(lang.message_need_register + lang.message_footer).render(username=msg.author.name))
def balance_user(rpc, msg, failover_time): if user_function.user_exist(msg.author.name): if time.time() > int(failover_time.value) + 86400: # not in safe mode balance = get_user_confirmed_balance(rpc, msg.author.name) spendable_balance = get_user_spendable_balance(rpc, msg.author.name) + balance else: # we are in safe mode spendable_balance = get_user_confirmed_balance(rpc, msg.author.name) return spendable_balance
def withdraw_user(rpc, msg, failover_time): split_message = msg.body.strip().split() if user_function.user_exist(msg.author.name): sender_address = user_function.get_user_address(msg.author.name) amount = float(split_message[1]) amount = round(amount - 0.5) print(amount) user_balance = crypto.get_user_confirmed_balance(rpc, msg.author.name) user_spendable_balance = crypto.get_user_spendable_balance( rpc, msg.author.name) if utils.check_amount_valid( amount) and split_message[4] != sender_address: if amount >= float(user_balance) + float(user_spendable_balance): bot_logger.logger.info( 'user %s not have enough to withdraw this amount (%s), balance = %s' % (msg.author.name, amount, user_balance)) msg.reply( Template(lang.message_balance_low_withdraw).render( username=msg.author.name, user_balance=str(user_balance), amount=str(amount)) + lang.message_footer) else: receiver_address = split_message[4] if time.time() > int(failover_time.value) + 86400: send = crypto.send_to(rpc, sender_address, receiver_address, amount) else: send = crypto.send_to_failover(rpc, sender_address, receiver_address, amount) if send: user_function.add_to_history(msg.author.name, sender_address, receiver_address, amount, "withdraw") value_usd = utils.get_coin_value(amount) msg.reply( Template(lang.message_withdraw + lang.message_footer).render( username=msg.author.name, receiver_address=receiver_address, amount=str(amount), value_usd=str(value_usd))) elif split_message[4] == sender_address: msg.reply(lang.message_withdraw_to_self + lang.message_footer) else: bot_logger.logger.info(lang.message_invalid_amount) msg.reply(lang.message_invalid_amount + lang.message_footer) else: msg.reply( Template(lang.message_need_register + lang.message_footer).render(username=msg.author.name))
def register_user(rpc, msg): if not user_function.user_exist(msg.author.name): address = rpc.getnewaddress() if address: msg.reply(msg.author.name + ' registered, your address is ' + address) user_function.add_user(msg.author.name, address) pending_tips(rpc, msg) else: print 'Error during register !' else: print msg.author.name + ' are already registered ' msg.reply('You are already registered ')
def replay_remove_pending_tip(rpc, reddit, tx_queue, failover_time): # check if it's not too old & replay tipping limit_date = datetime.datetime.now() - datetime.timedelta(days=3) # check if user have pending tips list_tips = user_function.get_unregistered_tip() if list_tips: for tip in list_tips: bot_logger.logger.info("replay tipping check for %s" % str(tip['id'])) if (datetime.datetime.strptime(tip['time'], '%Y-%m-%dT%H:%M:%S.%f') > limit_date): if (user_function.user_exist(tip['receiver'])): bot_logger.logger.info( "replay tipping %s - %s send %s to %s " % (str(tip['id']), tip['sender'], tip['amount'], tip['receiver'])) txid = crypto.tip_user(rpc, tip['sender'], tip['receiver'], tip['amount'], tx_queue, failover_time) user_function.remove_pending_tip(tip['id']) value_usd = utils.get_coin_value(tip['amount']) if 'message_fullname' in tip.keys(): msg_id = re.sub(r't\d+_(?P<id>\w+)', r'\g<id>', tip['message_fullname']) msg = Comment(reddit, msg_id) msg.reply( Template(lang.message_tip).render( sender=tip['sender'], receiver=tip['receiver'], amount=str(tip['amount']), value_usd=str(value_usd), txid=txid)) else: bot_logger.logger.info( "replay check for %s - user %s not registered " % (str(tip['id']), tip['receiver'])) else: bot_logger.logger.info( "delete old tipping - %s send %s for %s " % (tip['sender'], tip['amount'], tip['receiver'])) user_function.remove_pending_tip(tip['id']) else: bot_logger.logger.info("no pending tipping")
def history_user(msg): if user_function.user_exist(msg.author.name): data = user_function.get_user_history(msg.author.name) history_table = "\n\nDate|Sender|Receiver|Amount|Action|Finish|\n" history_table += "---|---|---|---|:-:|:-:\n" for tip in data: str_finish = "Pending" if tip['finish']: str_finish = "Successful" history_table += "%s|%s|%s|%s|%s|%s|\n" % ( datetime.datetime.strptime( tip['time'], '%Y-%m-%dT%H:%M:%S.%f'), tip['sender'], tip['receiver'], str(tip['amount']), tip['action'], str_finish) msg.reply( Template(lang.message_history + history_table + lang.message_footer).render(username=msg.author.name)) else: bot_logger.logger.info('user %s not registered ' % msg.author.name) msg.reply( Template(lang.message_need_register + lang.message_footer).render(username=msg.author.name))
def info_user(rpc, msg): if user_function.user_exist(msg.author.name): user_function.get_user_info(msg) else: msg.reply('You need %s before' % linkRegister)
def tip_user(rpc, reddit, msg, tx_queue, failover_time): bot_logger.logger.info('An user mention detected ') bot_logger.logger.debug("failover_time : %s " % (str(failover_time.value))) split_message = msg.body.lower().strip().split() tip_index = split_message.index(str('+/u/' + config.bot_name)) if split_message[tip_index] == str( '+/u/' + config.bot_name) and split_message[tip_index + 2] == 'doge': amount = float(split_message[tip_index + 1]) amount = round(amount - 0.5) if utils.check_amount_valid(amount): parent_comment = msg.parent() if user_function.user_exist(msg.author.name) and ( msg.author.name != parent_comment.author.name): # check we have enough user_balance = crypto.get_user_confirmed_balance( rpc, msg.author.name) user_pending_balance = crypto.get_user_unconfirmed_balance( rpc, msg.author.name) user_spendable_balance = crypto.balance_user( rpc, msg, failover_time) bot_logger.logger.debug('user_spendable_balance = %s' % user_spendable_balance) # in failover we need to use only user_balance if amount >= float(user_spendable_balance): # not enough for tip if amount < float(user_pending_balance): reddit.redditor(msg.author.name).message( 'pending tip', Template(lang.message_balance_pending_tip).render( username=msg.author.name)) else: bot_logger.logger.info( 'user %s not have enough to tip this amount (%s), balance = %s' % (msg.author.name, str(amount), str(user_balance))) reddit.redditor(msg.author.name).message( 'low balance', Template(lang.message_balance_low_tip).render( username=msg.author.name)) else: value_usd = utils.get_coin_value(amount) # check user have address before tip if user_function.user_exist(parent_comment.author.name): txid = crypto.tip_user(rpc, msg.author.name, parent_comment.author.name, amount, tx_queue, failover_time) if txid: user_function.add_to_history( msg.author.name, msg.author.name, parent_comment.author.name, amount, "tip send", txid) user_function.add_to_history( parent_comment.author.name, msg.author.name, parent_comment.author.name, amount, "tip receive", txid) bot_logger.logger.info( '%s tip %s to %s' % (msg.author.name, str(amount), parent_comment.author.name)) # if user have 'verify' in this command he will have confirmation if split_message.count( 'verify') or int(amount) >= 1000: msg.reply( Template(lang.message_tip).render( sender=msg.author.name, receiver=parent_comment.author.name, amount=str(int(amount)), value_usd=str(value_usd), txid=txid)) else: user_function.save_unregistered_tip( msg.author.name, parent_comment.author.name, amount, msg.fullname) user_function.add_to_history( msg.author.name, msg.author.name, parent_comment.author.name, amount, "tip send", False) user_function.add_to_history( parent_comment.author.name, msg.author.name, parent_comment.author.name, amount, "tip receive", False) bot_logger.logger.info('user %s not registered' % parent_comment.author.name) reddit.redditor(msg.author.name).message( 'tipped user not registered', Template(lang.message_recipient_register).render( username=parent_comment.author.name)) reddit.redditor(parent_comment.author.name).message( Template(lang.message_recipient_need_register_title ).render(amount=str(amount)), Template( lang.message_recipient_need_register_message). render(username=parent_comment.author.name, sender=msg.author.name, amount=str(amount), value_usd=str(value_usd))) elif user_function.user_exist( msg.author.name) and (msg.author.name == parent_comment.author.name): reddit.redditor(msg.author.name).message( 'cannot tip self', Template(lang.message_recipient_self).render( username=msg.author.name)) else: reddit.redditor(msg.author.name).message( 'tipped user not registered', Template(lang.message_need_register).render( username=msg.author.name)) else: bot_logger.logger.info(lang.message_invalid_amount) reddit.redditor(msg.author.name).message( 'invalid amount', lang.message_invalid_amount)
def __init__(self, user): self.username = user self.address = None if user_function.user_exist(self.username): self.address = user_function.get_user_address(self.username)