def get_sellers(): """ :return: """ cur = cnn.cursor() try: cur.execute("select seller_name from sellers order by id;") except: cnn.rollback() res = [item[0] for item in cur.fetchall()] cnn.commit() return res
def add_seller(seller_name): """ add seller to seller's table :param seller: name of seller :return: """ cur = cnn.cursor() try: sql = "INSERT INTO Sellers(seller_name) VALUES(%s);" cur.execute(sql, (seller_name, )) cnn.commit() except: cnn.rollback() raise
def unconnect_user_db(user_id, to_user_id): """ add pair of connected users into db. :param user_id: :param to_user_id: """ sql = "SELECT set_user_unconnect(%s, %s)" cur = cnn.cursor() try: cur.execute(sql, (user_id, to_user_id)) cnn.commit() et.info("pair of user ({0} - {1}) unconnected".format( user_id, to_user_id)) except: cnn.rollback() et.error("error during unconnect_user_db\n" + traceback.format_exc())
def connect_user_db(user_id, to_user_id): """ add pair of connected users into db. :param user_id: :param to_user_id: """ sql = "SELECT add_connected_user(%s, %s)" cur = cnn.cursor() try: cur.execute(sql, (user_id, to_user_id)) cnn.commit() et.info("pair of user ({0} - {1}) add".format(user_id, to_user_id)) except psycopg2.IntegrityError: cnn.rollback() et.info("pair of user ({0},{1}) already exists".format( user_id, to_user_id))
def get_users_to_unconnect_db(user_id): """ get to_users_id who didn't connect user_name. :param user_id: user, for whom users should be unconnected. """ et.info("START get_users_to_unconnect_db") cur = cnn.cursor() users = {} sql = "SELECT get_users_to_unconnect(%s)" try: cur.execute(sql, (user_id, )) users = cur.fetchall() cnn.commit() except: cnn.rollback() et.info("FINISH get_users_to_unconnect_db") return users
def was_user_connected_to_user(user_id, to_user_id): """ check if to_user_id was ever connected to user_id. :param user_id: :param to_user_id: :return boolean. """ et.info(msg="START was_user_connected_to_user();") sql = "SELECT was_user_connected_to_user(%s, %s)" cur = cnn.cursor() res = False try: cur.execute(sql, (user_id, to_user_id)) res = cur.fetchone()[0] cnn.commit() except: cnn.rollback() et.error("error during was_user_connected_to_user\n" + traceback.format_exc()) et.info(msg="FINISH was_user_connected_to_user();") return res