如何在psycopg2连接池中使用“插入”？

db = pool.SimpleConnectionPool(1, 10,host=conf_hostname,database=conf_dbname,user=conf_dbuser,password=conf_dbpass,port=conf_dbport)
# Get Cursor
@contextmanager
def get_cursor():
    con = db.getconn()
        yield con.cursor()
    finally:
        db.putconn(con)
with get_cursor() as cursor:
    cursor.execute("INSERT INTO table (fields) VALUES (values) RETURNING id") 
    id = cursor.fetchone()

db = pool.SimpleConnectionPool(1, 10,host=conf_hostname,database=conf_dbname,user=conf_dbuser,password=conf_dbpass,port=conf_dbport)
@contextmanager
def get_connection():
    con = db.getconn()
        yield con
    finally:
        db.putconn(con)
def write_to_db():
    with get_connection() as conn:
            cursor = conn.cursor()
            cursor.execute("INSERT INTO table (fields) VALUES (values) RETURNING id") 
            id = cursor.fetchone()
            cursor.close()
            conn.commit()
        except:
            conn.rollback()

db_pool = pool.SimpleConnectionPool(1, 10,
                                    host=CONF.db_host,
                                    database=CONF.db_name, 
                                    user=CONF.db_user, 
                                    password=CONF.db_user,
                                    port=CONF.db_port)
@contextmanager
def db():
    con = db_pool.getconn()
    cur = con.cursor()
        yield con, cur
    finally:
        cur.close()
        db_pool.putconn(con)
if __name__ == '__main__':
    with db() as (connection, cursor):
            cursor.execute("""INSERT INTO table (fields)
VALUES (values) RETURNING id""")
            my_id = cursor.fetchone()
            rowcount = cursor.rowcount
            if rowcount == 1:
                connection.commit()

推荐文章

活泼的打火机 · 五星泰山背后，是这些人的专业和细致|鲁能|山东泰山|足球_网易订阅

4 月前

安静的火柴 · Stomp on Spring WebSocket 有种端到Topic下指定用户通信的通信模式：给特定的用户发送消息，例如默认的/user前缀下，客户端可以订阅/user/topic/chat表明监听

9 月前

温文尔雅的橙子 · Android字符串资源及其格式化-阿里云开发者社区

9 月前

英勇无比的野马 · 关于新冠疫情影响下加大对餐饮行业的扶持力度的建议-建议提案办理-深圳市地方金融监督管理局(深圳市政府金融工作办公室)

1 年前

骑白马的菠菜 · Jenkins: Accessed denied after turning on global security. How to revert? - Server Fault

1 年前

如何在psycopg2连接池中使用“插入”？

2 个回答