百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 热门文章 > 正文

一日一技:Python | PostgreSQL中的数据库管理

bigegpt 2025-02-03 11:28 7 浏览

有几个python模块可以让我们使用PostgreSQL连接和操作数据库:


  1. Psycopg2
  2. pg8000
  3. py-postgresql
  4. PyGreSQL



Psycopg2是PostgreSQL最受欢迎的python驱动程序之一。 它被积极维护并为不同版本的python提供支持。 它还提供对线程的支持,并且可以在多线程应用程序中使用。 由于这些原因,它是开发人员的流行选择。


在这一小节中,我们将通过在python中构建一个简单的数据库管理系统来探索使用psycopg2使用PostgreSQl的功能。


安装模块:

sudo pip3 install psycopg2   #使用的是 Ubuntu系统

注意:如果您使用的是Python2,请使用pip install代替pip3,不过python2.7版本已经不再维护,不推荐使用。

在您的系统中安装了psycopg之后,我们可以连接到数据库并在Python中执行查询。


创建数据库

在我们可以使用python访问数据库之前,我们需要在postgresql中创建数据库。 要创建数据库,请遵循以下步骤:

1.登录PostgreSQL.

sudo -u postgres psql

2.配置密码.

\password

然后将提示您输入密码。 记住这一点,因为我们将使用它来连接到Python中的数据库。

3.创建一个名为“ test”的数据库。 我们将连接到该数据库.

CREATE DATABASE test;   #分号;别忘记带上

配置数据库和密码后,退出psql服务器。

连接到数据库

connect()方法用于建立与数据库的连接。 它包含5个参数:

1.database:您要连接的数据库的名称

2.user:您本地系统的用户名

3.password:登录psql的密码

4.host:主机,默认情况下设置为localhost

5.port:端口号,默认为5432



conn = psycopg2.connect(
            database="test", 
            user = "adith", 
            password = "password", 
            host = "localhost", 
            port = "5432")



建立连接后,我们可以使用python操作数据库。

Cursor对象用于执行sql查询。 我们可以使用连接对象(conn)创建一个游标对象

cur = conn.cursor()  

使用此对象,我们可以更改连接到的数据库



执行完所有查询后,我们需要断开连接。 不断开连接不会导致任何错误,但是通常认为断开连接是一种好习惯。

 conn.close() 

执行查询

execute()方法采用一个参数,即要执行的SQL查询。 SQL查询采用包含SQL语句的字符串形式。

cur.execute("SELECT * FROM emp") 



获得数据

一旦执行了查询,就可以使用fetchall()方法获取查询的结果。 此方法不带参数,并返回选择查询的结果。

 res = cur.fetchall() 

查询结果存储在res变量中.



全部放在一起

在PostgreSQL中创建数据库后,就可以使用python访问该数据库。 我们首先使用以下模式在数据库中创建一个名为test的emp表:(id INTEGER PRIMARY KEY,名称VARCHAR(10),salary INT,dept INT)。 创建表后,没有任何错误,我们将值插入表中。

插入值后,我们可以查询表以选择所有行,并使用fetchall()函数将其显示给用户。



# importing libraries 
import psycopg2 

# a function to connect to 
# the database. 
def connect(): 

	# connecting to the database called test 
	# using the connect function 
	try: 

		conn = psycopg2.connect(database ="test", 
							user = "adith", 
							password = "password", 
							host = "localhost", 
							port = "5432") 

		# creating the cursor object 
		cur = conn.cursor() 
	
	except (Exception, psycopg2.DatabaseError) as error: 
		
		print ("Error while creating PostgreSQL table", error) 
	

	# returing the conn and cur 
	# objects to be used later 
	return conn, cur 


# a function to create the 
# emp table. 
def create_table(): 

	# connect to the database. 
	conn, cur = connect() 

	try: 
		# the test database contains a table called emp 
		# the schema : (id INTEGER PRIMARY KEY, 
		# name VARCHAR(10), salary INT, dept INT) 
		# create the emp table 

		cur.execute('CREATE TABLE emp (id INT PRIMARY KEY, name VARCHAR(10), 
									salary INT, dept INT)') 

		# the commit function permanently 
		# saves the changes made to the database 
		# the rollback() function can be used if 
		# there are any undesirable changes and 
		# it simply undoes the changes of the 
		# previous query 
	
	except: 

		print('error') 

	conn.commit() 


# a function to insert data 
# into the emp table 
def insert_data(id = 1, name = '', salary = 1000, dept = 1): 

	conn, cur = connect() 

	try: 
		# inserting values into the emp table 
		cur.execute('INSERT INTO emp VALUES(%s, %s, %s, %s)', 
									(id, name, salary, dept)) 
	
	except Exception as e: 

		print('error', e) 
	# commiting the transaction. 
	conn.commit() 


# a function to fetch the data 
# from the table 
def fetch_data(): 

	conn, cur = connect() 

	# select all the rows from emp 
	try: 
		cur.execute('SELECT * FROM emp') 
	
	except: 
		print('error !') 

	# store the result in data 
	data = cur.fetchall() 

	# return the result 
	return data 

# a function to print the data 
def print_data(data): 

	print('Query result: ') 
	print() 

	# iterating over all the 
	# rows in the table 
	for row in data: 

		# printing the columns 
		print('id: ', row[0]) 
		print('name: ', row[1]) 
		print('salary: ', row[2]) 
		print('dept: ', row[3]) 
		print('----------------------------------') 

# function to delete the table 
def delete_table(): 

	conn, cur = connect() 

	# delete the table 
	try: 

		cur.execute('DROP TABLE emp') 

	except Exception as e: 
		print('error', e) 

	conn.commit() 


# driver function 
if __name__ == '__main__': 

	# create the table 

	create_table() 

	# inserting some values 
	insert_data(1, 'adith', 1000, 2) 
	insert_data(2, 'tyrion', 100000, 2) 
	insert_data(3, 'jon', 100, 3) 
	insert_data(4, 'daenerys', 10000, 4) 

	# getting all the rows 
	data = fetch_data() 

	# printing the rows 
	print_data(data) 

	# deleting the table 
	# once we are done with 
	# the program 
	delete_table() 

输出:

相关推荐

得物可观测平台架构升级:基于GreptimeDB的全新监控体系实践

一、摘要在前端可观测分析场景中,需要实时观测并处理多地、多环境的运行情况,以保障Web应用和移动端的可用性与性能。传统方案往往依赖代理Agent→消息队列→流计算引擎→OLAP存储...

warm-flow新春版:网关直连和流程图重构

本期主要解决了网关直连和流程图重构,可以自此之后可支持各种复杂的网关混合、多网关直连使用。-新增Ruoyi-Vue-Plus优秀开源集成案例更新日志[feat]导入、导出和保存等新增json格式支持...

扣子空间体验报告

在数字化时代,智能工具的应用正不断拓展到我们工作和生活的各个角落。从任务规划到项目执行,再到任务管理,作者深入探讨了这款工具在不同场景下的表现和潜力。通过具体的应用实例,文章展示了扣子空间如何帮助用户...

spider-flow:开源的可视化方式定义爬虫方案

spider-flow简介spider-flow是一个爬虫平台,以可视化推拽方式定义爬取流程,无需代码即可实现一个爬虫服务。spider-flow特性支持css选择器、正则提取支持JSON/XML格式...

solon-flow 你好世界!

solon-flow是一个基础级的流处理引擎(可用于业务规则、决策处理、计算编排、流程审批等......)。提供有“开放式”驱动定制支持,像jdbc有mysql或pgsql等驱动,可...

新一代开源爬虫平台:SpiderFlow

SpiderFlow:新一代爬虫平台,以图形化方式定义爬虫流程,不写代码即可完成爬虫。-精选真开源,释放新价值。概览Spider-Flow是一个开源的、面向所有用户的Web端爬虫构建平台,它使用Ja...

通过 SQL 训练机器学习模型的引擎

关注薪资待遇的同学应该知道,机器学习相关的岗位工资普遍偏高啊。同时随着各种通用机器学习框架的出现,机器学习的门槛也在逐渐降低,训练一个简单的机器学习模型变得不那么难。但是不得不承认对于一些数据相关的工...

鼠须管输入法rime for Mac

鼠须管输入法forMac是一款十分新颖的跨平台输入法软件,全名是中州韵输入法引擎,鼠须管输入法mac版不仅仅是一个输入法,而是一个输入法算法框架。Rime的基础架构十分精良,一套算法支持了拼音、...

Go语言 1.20 版本正式发布:新版详细介绍

Go1.20简介最新的Go版本1.20在Go1.19发布六个月后发布。它的大部分更改都在工具链、运行时和库的实现中。一如既往,该版本保持了Go1的兼容性承诺。我们期望几乎所...

iOS 10平台SpriteKit新特性之Tile Maps(上)

简介苹果公司在WWDC2016大会上向人们展示了一大批新的好东西。其中之一就是SpriteKitTileEditor。这款工具易于上手,而且看起来速度特别快。在本教程中,你将了解关于TileE...

程序员简历例句—范例Java、Python、C++模板

个人简介通用简介:有良好的代码风格,通过添加注释提高代码可读性,注重代码质量,研读过XXX,XXX等多个开源项目源码从而学习增强代码的健壮性与扩展性。具备良好的代码编程习惯及文档编写能力,参与多个高...

Telerik UI for iOS Q3 2015正式发布

近日,TelerikUIforiOS正式发布了Q32015。新版本新增对XCode7、Swift2.0和iOS9的支持,同时还新增了对数轴、不连续的日期时间轴等;改进TKDataPoin...

ios使用ijkplayer+nginx进行视频直播

上两节,我们讲到使用nginx和ngixn的rtmp模块搭建直播的服务器,接着我们讲解了在Android使用ijkplayer来作为我们的视频直播播放器,整个过程中,需要注意的就是ijlplayer编...

IOS技术分享|iOS快速生成开发文档(一)

前言对于开发人员而言,文档的作用不言而喻。文档不仅可以提高软件开发效率,还能便于以后的软件开发、使用和维护。本文主要讲述Objective-C快速生成开发文档工具appledoc。简介apple...

macOS下配置VS Code C++开发环境

本文介绍在苹果macOS操作系统下,配置VisualStudioCode的C/C++开发环境的过程,本环境使用Clang/LLVM编译器和调试器。一、前置条件本文默认前置条件是,您的开发设备已...