import sqlite3
from sqlite3 import Error
import logging


logger = logging.getLogger(__name__)
logging.FileHandler('logfile.log')
logging.basicConfig(filename='logfile.log', filemode='w', level=logging.INFO,
                    format='%(asctime)s :: %(levelname)s :: %(name)s :: %(message)s')


class DbConnect:

    def __init__(self, db_path):
        self.connection = None
        self.db_path = db_path

    def get_connection(self):
        try:
            self.connection = sqlite3.connect(self.db_path)
            logger.info("database connected successfully")
        except Error as e:
            logger.critical(f"error: {e}")
            # print(f"An Error has occurred: {e}")
        # return connection

    def run_query(self, sql_query, row):
        cursor = self.connection.cursor()
        try:
            cursor.execute(sql_query, row)
            self.connection.commit()
            logger.debug(sql_query)
            logger.debug(row)
            logger.info("SQL query run successfully")
        except Error as e:
            print(f" Query Failed……{e}")

    def get_data(self, query):
        col = []
        cursor = self.connection.cursor()
        cursor.execute(query)
        rows = cursor.fetchall()
        for row in rows:
            col.append(row)
        return col

    def get_notready(self):
        col = []
        cursor = self.connection.cursor()
        cursor.execute("""SELECT
                            name, city_id FROM cities 
                            WHERE city_id NOT IN 
                            (SELECT city_id FROM olx_data 
                            WHERE date=date('now'));            
                        """)
        rows = cursor.fetchall()
        for row in rows:
            col.append(row)
        return col


# con = DbConnect("/home/krzychu/PycharmProjects/OLX_selenium/test_table.db")
# con.get_connection()
# regions = con.get_column("regions", "name, region_id")
# cities = con.get_column("cities", "city_id, name, city_region_id")

# cities_div = {}
#
# for el in regions:
#     # print(el)
#     cities_list = []
#     for i in cities:
#         if i[2] == el[1]:
#             cities_list.append(i[1])
#     cities_div[el[0]] = cities_list
#
# print(cities_div)

# print(regions)
# print(cities)

# add_record = """
# INSERT INTO
#    olx_data(city_id, adv_sale_count, adv_rent_count, adv_exchange_count)
# VALUES
#    (4548, 629, 837, 7);
# """

# run_query(connection=con, sql_query=add_record)