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

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

wptr33 2025-01-06 15:48 20 浏览

有几个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() 

输出:

相关推荐

SQL轻松入门(5):窗口函数(sql语录中加窗口函数的执行)

01前言标题中有2个字让我在初次接触窗口函数时,真真切切明白了何谓”高级”?说来也是一番辛酸史!话说,我见识了窗口函数的强大后,便磨拳擦掌的要试验一番,结果在查询中输入语句,返回的结果却是报错,Wh...

28个SQL常用的DeepSeek提示词指令,码住直接套用

自从DeepSeek出现后,极大地提升了大家平时的工作效率,特别是对于一些想从事数据行业的小白,只需要掌握DeepSeek的提问技巧,SQL相关的问题也不再是个门槛。...

从零开始学SQL进阶,数据分析师必备SQL取数技巧,建议收藏

上一节给大家讲到SQL取数的一些基本内容,包含SQL简单查询与高级查询,需要复习相关知识的同学可以跳转至上一节,本节给大家讲解SQL的进阶应用,在实际过程中用途比较多的子查询与窗口函数,下面一起学习。...

SQL_OVER语法(sql语句over什么含义)

OVER的定义OVER用于为行定义一个窗口,它对一组值进行操作,不需要使用GROUPBY子句对数据进行分组,能够在同一行中同时返回基础行的列和聚合列。...

SQL窗口函数知多少?(sql窗口怎么执行)

我们在日常工作中是否经常会遇到需要排名的情况,比如:每个部门按业绩来排名,每人按绩效排名,对部门销售业绩前N名的进行奖励等。面对这类需求,我们就需要使用sql的高级功能——窗口函数。...

如何学习并掌握 SQL 数据库基础:从零散查表到高效数据提取

无论是职场数据分析、产品运营,还是做副业项目,掌握SQL(StructuredQueryLanguage)意味着你能直接从数据库中提取、分析、整合数据,而不再依赖他人拉数,节省大量沟通成本,让你...

SQL窗口函数(sql窗口函数执行顺序)

背景在数据分析中,经常会遇到按某某条件来排名、并找出排名的前几名,用日常SQL的GROUPBY,ORDERBY来实现特别的麻烦,有时甚至实现不了,这个时候SQL窗口函数就能发挥巨大作用了,窗...

sqlserver删除重复数据只保留一条,使用ROW_NUMER()与Partition By

1.使用场景:公司的小程序需要实现一个功能:在原有小程序上,有一个优惠券活动表。存储着活动产品数据,但因为之前没有做约束,导致数据的不唯一,这会使打开产品详情页时,可能会出现随机显示任意活动问题。...

SQL面试经典问题(一)(sql经典面试题及答案)

以下是三个精心挑选的经典SQL面试问题及其详细解决方案,涵盖了数据分析、排序限制和数据清理等常见场景。这些问题旨在考察SQL的核心技能,适用于初学者到高级开发者的面试准备。每个问题均包含清晰的...

SQL:求连续N天的登陆人员之通用解答

前几天发了一个微头条:...

SQL四大排序函数神技(sql中的排序是什么语句)

在日常SQL开发中,排序操作无处不在。当大家需要排序时,是否只会想到ORDERBY?今天,我们就来揭秘SQL中四个强大却常被忽略的排序函数:ROW_NUMBER()、RANK()、DENSE_RAN...

四、mysql窗口函数之row_number()函数的使用

1、窗口函数之row_number()使用背景窗口函数中,排序函数rank(),dense_rank()虽说都是排序函数,但是各有用处,假如像上章节说的“同组同分”两条数据,我们不想“班级名次”出现“...

ROW_NUMBER()函数(rownumber函数与rank区别)

ROW_NUMBER()是SQL中的一个窗口函数(WindowFunction)...

Dify「模板转换」节点终极指南:动态文本生成进阶技巧(附代码)Jinja2引擎解析

这篇文章是关于Dify「模板转换」节点的终极指南,解析了基于Jinja2模板引擎的动态文本生成技巧,涵盖多源文本整合、知识检索结构化、动态API构建及个性化内容生成等六大应用场景,助力开发者高效利用模...

Python 最常用的语句、函数有哪些?

1.#coding=utf-8①代码中有中文字符,最好在代码前面加#coding=utf-8②pycharm不加可能不会报错,但是代码最终是会放到服务器上,放到服务器上的时候运行可能会报错。③...