def add_address(self,wallet_id,address): try: db = Database() conn = db.get_connection() cursor = db.get_cursor() if cursor == 0: return 0 except: return 0 if isinstance(wallet_id,int): cursor.execute(""" INSERT INTO `addresses_table`(wallet_id, address) VALUES ({},'{}') """.format(wallet_id,address)) conn.commit() else: return 0
def get_wallets(self): try: db = Database() conn = db.get_connection() cursor = db.get_cursor() if cursor == 0: return 0 except: return 0 status = cursor.execute(""" SELECT * FROM `wallets_table` """) if status > 0: data = cursor.fetchall() return data else: return 0
def get_wallet(self,wallet_id): try: db = Database() conn = db.get_connection() cursor = db.get_cursor() if cursor == 0: return 0 except: return 0 status = cursor.execute(""" SELECT * FROM `wallets_table` WHERE wallet_id={}""".format(int(wallet_id))) if status > 0: data = cursor.fetchone() return data else: return 0
def get_wallet_addresses(self, wallet_id): try: db = Database() conn = db.get_connection() cursor = db.get_cursor() # Note : Ordinary List Cursor if cursor == 0: return 0 except: return 0 address_list = [] if isinstance(wallet_id,int): cursor.execute(""" SELECT address FROM `addresses_table` WHERE wallet_id={}""".format(wallet_id)) data = cursor.fetchall() for i in data: address_list.append(i['address']) return address_list else: return 0
def add_wallet(self, wallet_name, wallet_desc, wallet_type): try: db = Database() conn = db.get_connection() cursor = db.get_cursor() if cursor == 0: return 0 except: return 0 wallet_id = int(str(random.randint(1,9)) + str(random.randint(1,9)) + str(random.randint(1,9)) + str(random.randint(1,9)) + str(random.randint(1,9)) + str(random.randint(1,9))) if isinstance(wallet_name,str) and isinstance(wallet_desc,str) and isinstance(wallet_type,str): if len(wallet_name) > 2 and len(wallet_type) > 1: status = cursor.execute("""INSERT INTO `wallets_table`(wallet_id, wallet_name, wallet_desc, wallet_type) VALUES ('{}','{}','{}','{}')""".format(wallet_id,wallet_name,wallet_desc,wallet_type)) conn.commit() return 1 else: return 0
def update_balance(self, wallet_id, balance): try: db = Database() conn = db.get_connection() cursor = db.get_cursor() if cursor == 0: return 0 except: return 0 if isinstance(balance,float) and isinstance(wallet_id,int): status = cursor.execute("UPDATE `wallets_table` SET wallet_balance={} WHERE wallet_id={}".format(balance,wallet_id)) conn.commit() if status == 0: return 0 else: return 1 else: return 0