mirror of
https://github.com/ada-dmitry/lab6.git
synced 2026-09-24 09:20:15 +00:00
create 0.8.1
This commit is contained in:
@@ -0,0 +1,34 @@
|
||||
# Установка соединения с базой данных
|
||||
# (параметры передаются через класс конфиг).
|
||||
import psycopg2
|
||||
|
||||
class DbConnection:
|
||||
|
||||
def __init__(self, config):
|
||||
self.dbname = config.dbname
|
||||
self.user = config.user
|
||||
self.password = config.password
|
||||
self.host = config.host
|
||||
self.prefix = config.dbtableprefix
|
||||
self.conn = psycopg2.connect(dbname = self.dbname,
|
||||
user = self.user,
|
||||
password = self.password,
|
||||
host = self.host)
|
||||
|
||||
def __del__(self):
|
||||
if self.conn:
|
||||
self.conn.close()
|
||||
|
||||
def test(self):
|
||||
cur = self.conn.cursor()
|
||||
cur.execute("DROP TABLE IF EXISTS test CASCADE")
|
||||
cur.execute("CREATE TABLE test(test integer)")
|
||||
cur.execute("INSERT INTO test(test) VALUES(1)")
|
||||
self.conn.commit()
|
||||
cur.execute("SELECT * FROM test")
|
||||
result = cur.fetchall()
|
||||
cur.execute("DROP TABLE test")
|
||||
self.conn.commit()
|
||||
return (result[0][0] == 1)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user