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()