91 lines
2.3 KiB
Python
91 lines
2.3 KiB
Python
import psycopg2
|
|
|
|
class Postgres:
|
|
def __init__(self):
|
|
self.db_ip = 'localhost'
|
|
self.db_port = '32771'
|
|
self.db_username = 'root'
|
|
self.db_password = ''
|
|
self.db_name = ''
|
|
|
|
def config(self, ip, port, username, password, name):
|
|
self.db_ip = ip
|
|
self.db_port = port
|
|
self.db_username = username
|
|
self.db_password = password
|
|
self.db_name = name
|
|
|
|
def fetchall(self, query):
|
|
conn = psycopg2.connect(
|
|
host=self.db_ip,
|
|
port=self.db_port,
|
|
user=self.db_username,
|
|
password=self.db_password,
|
|
database=self.db_name
|
|
)
|
|
cur = conn.cursor()
|
|
cur.execute(query)
|
|
result = cur.fetchall()
|
|
cur.close()
|
|
conn.close()
|
|
return result
|
|
|
|
def fetchone(self, query):
|
|
conn = psycopg2.connect(
|
|
host=self.db_ip,
|
|
port=self.db_port,
|
|
user=self.db_username,
|
|
password=self.db_password,
|
|
database=self.db_name
|
|
)
|
|
cur = conn.cursor()
|
|
cur.execute(query)
|
|
result = cur.fetchone()
|
|
cur.close()
|
|
conn.close()
|
|
return result
|
|
|
|
def fetchmany(self, query, size):
|
|
conn = psycopg2.connect(
|
|
host=self.db_ip,
|
|
port=self.db_port,
|
|
user=self.db_username,
|
|
password=self.db_password,
|
|
database=self.db_name
|
|
)
|
|
cur = conn.cursor()
|
|
cur.execute(query)
|
|
result = cur.fetchmany(size)
|
|
cur.close()
|
|
conn.close()
|
|
return result
|
|
|
|
def execute(self, query):
|
|
conn = psycopg2.connect(
|
|
host=self.db_ip,
|
|
port=self.db_port,
|
|
user=self.db_username,
|
|
password=self.db_password,
|
|
database=self.db_name
|
|
)
|
|
cur = conn.cursor()
|
|
cur.execute(query)
|
|
conn.commit()
|
|
cur.close()
|
|
conn.close()
|
|
|
|
def executemany(self, query, values):
|
|
conn = psycopg2.connect(
|
|
host=self.db_ip,
|
|
port=self.db_port,
|
|
user=self.db_username,
|
|
password=self.db_password,
|
|
database=self.db_name
|
|
)
|
|
cur = conn.cursor()
|
|
cur.executemany(query, values)
|
|
conn.commit()
|
|
cur.close()
|
|
conn.close()
|
|
|