如何在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()

推荐文章

高大的红薯 · 从零开始编译一个gcc的交叉编译工具链 - 周荣华 - 博客园

12 月前

瘦瘦的打火机 · PHP 3、array_merge和array_values重排索引数组性能比较 - 简书

1 年前

聪明伶俐的红薯 · C#反射与特性(二)：探究反射 - 腾讯云开发者社区-腾讯云

1 年前

温柔的木耳 · kubernetes - Failed create pod sandbox: rpc error: code = Unknown desc = NetworkPlugin cni failed to set up pod network - Stack Ov

1 年前

气宇轩昂的啄木鸟 · 解决方案:Can't process attribute android:fillColor="@android:color/white":_技术宅星云的博客-CSDN博客

1 年前

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

2 个回答