db.py

import psycopg2 as pg
from config import host, user, password, db_name

with pg.connect(
        host=host,  # 127.0.0.1
        user=user,  # postgres
        password=password,  # postgres
        database=db_name  # postgres
) as conn:
    conn.autocommit = True


def create_table_seen_person():  # references users(id_vk)
    """create table seen_person"""
    with conn.cursor() as cursor:
        cursor.execute(
            """CREATE TABLE IF NOT EXISTS seen_person(
            id serial,
            id_vk varchar(50) PRIMARY KEY);"""
        )


def insert_data_seen_person(id_vk):
    """inserting data into the seen_users table"""
    with conn.cursor() as cursor:
        cursor.execute(
            f"""INSERT INTO seen_person (id_vk) 
           VALUES (%s)""",
            (id_vk,)
        )


def check():
    with conn.cursor() as cursor:
        cursor.execute(
            f"""SELECT sp.id_vk
            FROM seen_person AS sp;"""
        )
        return cursor.fetchall()


def delete_table_seen_person():
    """delete table seen_person by cascade"""
    with conn.cursor() as cursor:
        cursor.execute(
            """DROP TABLE  IF EXISTS seen_person CASCADE;"""
        )



create_table_seen_person()
print("Database was created!")