client - master - valek/clients-db

client
61 строка2 KB
valek
create client
22 янв 2026, 11:33
22 янв 2026, 11:3387ffdc7
import psycopg2


def create_db(conn):
    with conn.cursor() as cur:
        cur.execute("""CREATE TABLE If NOT EXISTS clients
                (id serial PRIMARY KEY,
                name VARCHAR(60) NOT NULL,
                last_name VARCHAR(60) NOT NULL,
                email TEXT NOT NULL);
                """)
        cur.execute("""CREATE TABLE If NOT EXISTS phones
                (id serial PRIMARY KEY,
                client_id INTEGER NOT NULL REFERENCES clients(id),
                phone TEXT NOT NULL);
                """)
        return conn.commit()

def add_client(conn, name, last_name, email):
    with conn.cursor() as cur:
        cur.execute("""INSERT INTO clients (name, last_name, email)
                VALUES (%s, %s, %s);""",(name, last_name, email))
        return conn.commit()

def add_phone(conn, client_id, phone):
    with conn.cursor() as cur:
        cur.execute("""INSERT INTO phones (client_id, phone)
                VALUES (%s, %s);""",(client_id, phone))
        return conn.commit()

def change_client(conn, id, name, last_name, email):
    with conn.cursor() as cur:
        cur.execute("""UPDATE clients SET name = %s, last_name = %s, email = %s
                WHERE id = %s;""",(name, last_name, email, id))
        return conn.commit()

def delete_phone(conn, client_id, phone):
    with conn.cursor() as cur:
        cur.execute("""DELETE FROM phones WHERE client_id = %s AND phone = %s;""",(client_id, phone))
        return conn.commit()

def delete_client(conn, id, client_id):
    if id:
        with conn.cursor() as cur:
            cur.execute("""DELETE FROM clients WHERE id = %s;""",(id,))
            return conn.commit()
    elif client_id:
        with conn.cursor() as cur:
            cur.execute("""DELETE FROM phones WHERE client_id = %s;""",(client_id,))
            return conn.commit()

def find_client(conn, id):
    with conn.cursor() as cur:
        cur.execute("""SELECT * FROM clients WHERE id = %s;""",(id,))
        print(cur.fetchall())


with psycopg2.connect(database="client_db", user="postgres", password="Nhfycytanm-,fknbrf-2026") as conn:
    create_db(conn)

conn.close()