init

My_bd/dbconnection.py

@@ -0,0 +1,61 @@
+import psycopg2
+
+class DbConnection:
+    '''config это класс, который находится в файле config.yaml'''
+    def __init__(self, config):
+        '''это метод, который вызывается при создании объекта на основе класса DbConnection, 
+            и в нём задаются все ключевые параметры'''
+        self.dbname = config.dbname
+        self.user = config.user
+        self.password = config.password
+        self.host = config.host
+        self.prefix = config.dbtableprefix
+        ''' conn - это переменная/атрибут'''
+        ''' self.conn - это мы создаем атрибутт conn для ЭТОГО объекта и он хранит информацию о соединении с бд'''
+        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 dif_stations(self):
+    #     '''cursor - это посредник между питоном и бд'''
+    #     cur = self.conn.cursor()
+    #     cur.execute("DROP TABLE IF EXISTS stations CASCADE")
+    #     cur.execute("CREATE TABLE stations(st_id serial, st_name varchar(60), tarrif_zone_id integer, st_index integer)")
+    #     cur.execute("INSERT INTO stations(st_id, st_name, tarrif_zone_id, st_index) VALUES(1, 'Нахабино', 1,1)")
+    #     self.conn.commit()
+    #     cur.execute("SELECT * FROM stations")
+    #     result = cur.fetchall()
+    #     cur.execute("DROP TABLE stations")
+    #     self.conn.commit()
+    #     return (result[0][0] == 1)
+
+    # def dif_routes(self):
+    #     cur = self.conn.cursor()
+    #     cur.execute("DROP TABLE IF EXISTS routes CASCADE")
+    #     cur.execute("CREATE TABLE routes(route_id serial, first_st_id integer, last_st_id integer)")
+    #     cur.execute("INSERT INTO routes(route_id, first_st_id, last_st_id) VALUES(1,1,2)")
+    #     self.conn.commit()
+    #     cur.execute("SELECT * FROM routes")
+    #     result = cur.fetchall()
+    #     cur.execute("DROP TABLE routes")
+    #     self.conn.commit()
+    #     return (result[0][0] == 1)
+        
+    def test(self):
+        '''cursor - это посредник между питоном и бд'''
+        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)
+