NJTaxAssessment_v2.py

import os
import sys
import shelve
import re
import shutil
import time
import pandas as pd
import send2trash
import threading
import logging
import requests
from UniversalFunctions import UniversalFunctions
import selenium
from datetime import datetime
from datetime import timedelta
from zipfile import ZipFile
from traceback import format_tb
from bs4 import BeautifulSoup
from seleniumwire import webdriver as sw_webdriver
from selenium import webdriver
from selenium.webdriver.edge.service import Service
from selenium.webdriver.edge.options import Options
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import ElementNotVisibleException
from selenium.common.exceptions import ElementNotSelectableException
from selenium.common.exceptions import ElementNotInteractableException
from selenium.common.exceptions import InvalidArgumentException
from selenium.common.exceptions import NoSuchAttributeException
from selenium.common.exceptions import NoSuchDriverException
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import WebDriverException
# from importlib import reload
# Allows for Selenium to click a button
from selenium.webdriver.support.select import Select
# sys.path.append('F:\\Python 2.0\\Projects\\Real Life Projects\\Universal_Functions')
# UniversalFunctions = __import__("UniversalFunctions")
# reload(UniversalFunctions)


class NJTaxAssessment:

    tax_assessment = 'https://tax1.co.monmouth.nj.us/cgi-bin/prc6.cgi?menu=index&ms_user=monm&district=1301&mode=11'
    coordinates = 'https://www.latlong.net/'
    state_county_dict = None

    def __init__(self, city=None, county=None):
        # What information do I need to initialize an instance of this class?
        NJTaxAssessment.load_state_dictionary()
        try:

            if (city and county) is not None:
                self._city = city.upper()
                self._county = county.upper()
                self._database = NJTaxAssessment.city_database(self._county,self._city)
            elif (city and county) is None:
                self._city = None
                self._county = None
                self._database = None
            else:
                raise AssertionError

        except AssertionError:
            print('Error!\nCity and County have to equal values or equal None')

    """ 
    ______________________________________________________________________________________________________________
                                    Use this section to house the decorator functions
    ______________________________________________________________________________________________________________
    """

    @staticmethod
    def logger_decorator(original_function):
        def wrapper(*args, **kwargs):
            logger = logging.getLogger(original_function.__name__)
            logger.setLevel(logging.DEBUG)
            logger.propagate = False
            # Create the FileHandler() and StreamHandler() loggers
            f_handler = logging.FileHandler(
                original_function.__name__ + ' ' + str(datetime.today().date()) + '.log')
            f_handler.setLevel(logging.DEBUG)
            c_handler = logging.StreamHandler()
            c_handler.setLevel(logging.INFO)
            # Create formatting for the loggers
            formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                                          datefmt='%d-%b-%y %H:%M:%S')
            # Set the formatter for each handler
            f_handler.setFormatter(formatter)
            c_handler.setFormatter(formatter)
            logger.addHandler(f_handler)
            logger.addHandler(c_handler)

            kwargs['logger'] = logger
            kwargs['f_handler'] = f_handler
            kwargs['c_handler'] = c_handler

            result = original_function(*args, **kwargs)

            if result is None:
                pass
            else:
                return result

        return wrapper

    @staticmethod
    def run_main(original_function):
        def wrapper(*args, **kwargs):
            pass
            # Formulate all the date variables
            # todays_date = datetime.datetime.today().date()
            # data_avail = Scraper.current_data
            # temp_date = str(todays_date).split('-')
            # day = int(temp_date[2])
            # month = int(temp_date[1])
            # year = temp_date[0]
            # current_run_date = datetime.datetime.strptime(year + '-' + temp_date[1] + '-' + '24', "%Y-%m-%d").date()
            #
            # # Logic for calculating the next date to run main()
            # if day < 24:
            #     next_run_date = year + '-' + temp_date[1] + '-' + '24'
            # elif day >= 24:
            #     if data_avail == Scraper.event_log[obj.no_of_runs - 1]['Latest Available Data']:
            #         next_run_date = year + '-' + temp_date[1] + '-' + '24'
            #     else:
            #         if month in [1, 2, 3, 4, 5, 6, 7, 8]:
            #             nm = str(month + 1)
            #             next_month = '0' + nm
            #             next_run_date = year + '-' + next_month + '-' + '24'
            #         elif month in [9, 10, 11]:
            #             next_month = str(month + 1)
            #             next_run_date = year + '-' + next_month + '-' + '24'
            #         elif month == 12:
            #             next_month = '01'
            #             year = str(int(temp_date[0]) + 1)
            #             next_run_date = year + '-' + next_month + '-' + '24'
            #
            # next_run_date = datetime.datetime.strptime(next_run_date, "%Y-%m-%d").date()
            # if todays_date >= current_run_date:
            #     if data_avail == Scraper.event_log[Scraper.no_of_runs - 1]['Latest Available Data']:
            #         sleep_time = timedelta(days=1)
            #         Scraper.waiting(sleep_time)
            #
            #         return 'RESTART'
            #
            #     else:
            #         good_to_go = original_function(*args, **kwargs)
            #
            #     return good_to_go
            #
            # elif current_run_date < todays_date < next_run_date:
            #     if todays_date < next_run_date:
            #         sleep_time = next_run_date - todays_date
            #         Scraper.waiting(sleep_time)
            #
            #         return 'RESTART'

        return wrapper

    """ 
    ______________________________________________________________________________________________________________
                            Use this section to house the instance, class and static functions
    ______________________________________________________________________________________________________________
    """

    def auction_locations(self, city):
        # Use the GSMLS Tax statesinfo website to scrap all the NJ Counties Foreclosure Auction Locations
        pass

    @property
    def city(self):
        return self._city

    @city.setter
    def city(self, county, value):
        self._city = value.upper()
        self._database = NJTaxAssessment.city_database(county, value)

    @property
    def county(self):
        return self._county

    @county.setter
    def county(self, value):
        self._county = value.upper()
        self._database = NJTaxAssessment.city_database(self.county, value)

    @property
    def database(self):
        return self._database

    @database.setter
    def database(self,county, value):
        self._database = NJTaxAssessment.city_database(county, value)

    # @staticmethod
    # def property_tax_legend():
    #     # Produce the NJ Property tax system legend database to be used
    #     # to give greater detail on what type of property is at the target address
    #
    #     filename = ''
    #     db = pd.read_csv(filename)
    #
    #     return db

    @staticmethod
    @logger_decorator
    def all_county_scrape(key, counties, driver_var, county_name=None, city_name=None, **kwargs):
        """

        :param key:
        :param counties:
        :param driver_var:
        :param county_name:
        :param city_name:
        :param kwargs:
        :return:
        """

        logger = kwargs['logger']
        f_handler = kwargs['f_handler']
        c_handler = kwargs['c_handler']

        # started_threads = []

        nj_tax_base_url = 'https://tax1.co.monmouth.nj.us/'

        try:
            county_option = WebDriverWait(driver_var, 5).until(
                EC.presence_of_element_located(
                    (By.XPATH, "//*[@id='normdiv']/form/table[1]/tbody/tr[3]/td[2]/select/option["
                     + str(key) + "]")))  # Make sure the XPATH includes the variable j
            logger.info(f'Now downloading the databases for {counties[key]} County')
            county_option.click()  # Step 1: Select the county
            results1 = driver_var.page_source
            cities = NJTaxAssessment.find_cities(results1)

            if (county_name and city_name) is None:
                with requests.Session() as session:
                    for k in cities.keys():
                        try:
                            city_option = driver_var.find_element(By.XPATH,
                                                                  "//*[@id='normdiv']/form/table[1]/tbody/tr[4]/td[2]/select/option["
                                                                  + str(k) + "]")  # Make sure the XPATH includes the variable k
                            results = NJTaxAssessment.database_check(counties[key], cities[k], logger)
                            if results == 'File Already Exists':
                                continue
                            else:
                                city_option.click()
                                output_option = driver_var.find_element(By.XPATH,
                                                                        "//*[@id='normdiv']/form/table[1]/tbody/tr[6]/td[2]/select")
                                output_option.click()  # Step 2: Select the city
                                excel_option = driver_var.find_element(By.XPATH,
                                                                       "//*[@id='normdiv']/form/table[1]/tbody/tr[6]/td[2]/select/option[3]")
                                excel_option.click()  # Step 3: Select Excel/CSV as the file output
                                submit_search = driver_var.find_element(By.XPATH,
                                                                        "//*[@id='normdiv']/form/table[3]/tbody/tr[2]/td[2]/input[1]")
                                submit_search.click()  # Step 4: Click the submit button to generate the zip file
                                time.sleep(1) # Give the page time to fully load. Have had instances where download link isn't captured
                                results2 = driver_var.page_source
                                download_link = NJTaxAssessment.find_zipfile(results2)
                                NJTaxAssessment.stream_zipfile(nj_tax_base_url, download_link, session)
                                NJTaxAssessment.unzip_and_extract(counties[key], cities[k], download_link.split('/')[2])
                                logger.info(f'The download for {cities[k]} has finished...')
                                driver_var.back()  # Step &: Go back to the previous page to start the loop over again
                                # Find out how to go back. The new search button doesn't work

                        except WebDriverException as WDE:
                            logger.exception(f'{WDE}')

            elif county_name is not None and city_name is None:
                pass

            elif (county_name and city_name) is not None:
                pass

        except ElementNotVisibleException as ENV:  # Make more specific exception handling blocks later
            logger.exception(f'{ENV}')

        except ElementNotSelectableException as ENS:
            logger.exception(f'{ENS}')

        except InvalidArgumentException as IAE:
            logger.exception(f'{IAE}')

        except NoSuchAttributeException as NSAE:
            logger.exception(f'{NSAE}')

        except NoSuchDriverException as NSDE:
            logger.exception(f'{NSDE}')

        except NoSuchElementException as NSEE:
            logger.exception(f'{NSEE}')

        except WebDriverException as WDE:
            logger.exception(f'{WDE}')

        else:

            # for threadobj in started_threads:
            #     threadobj.join()

            logger.info(f'All databases for {counties[key]} County have been downloaded')
            logger.removeHandler(f_handler)
            logger.removeHandler(c_handler)
            logging.shutdown()

    @staticmethod
    def city_database(county, city):
        """

        :param county:
        :param city:
        :return:
        """
        # produce the property database for the specified city

        base_path = 'F:\\Real Estate Investing\\JQH Holding Company LLC\\Property Data'
        county = county.upper()
        city = city.upper()
        target_path = os.path.join(base_path, county, city)
        city_list = os.listdir(target_path)

        try:
            if county == 'ESSEX':

                for i in city_list:
                    if city not in i:
                        continue
                    elif city in i:
                        filename = os.path.join(target_path, i + '.xlsx')
                        db = pd.read_excel(filename, header=0)
                    else:
                        raise IndexError(f'{city} does not exist in {county} County')
            else:
                for idx, municipality in enumerate(city_list):
                    # This code could present future errors for counties which have duplicate city spellings. Fix this
                    if city not in municipality:
                        continue
                    elif city in municipality:
                        filename = os.path.join(target_path, city_list[idx])
                        db = pd.read_csv(filename, header=0)
                    else:
                        raise IndexError(f'{city} does not exist in {county} County')
        except IndexError as IE:
            print(f'{IE}')
        else:

            # Pre-clean the tax_db for use in GSMLS.py

            db.insert(0, 'Property Location', db.pop('Property Location').str.replace(r'RD$|RD\.$', 'ROAD', regex=True)
                      .str.replace(r'CT$|CT\.$', 'COURT', regex=True)
                      .str.replace(r'ST$|ST\.$', 'STREET', regex=True)
                      .str.replace(r'AVE$|AV$|AVE\.$|AV\.$', 'AVENUE', regex=True)
                      .str.replace(r'DR$|DR\.$', 'DRIVE', regex=True)
                      .str.replace(r'LN$|LA$|LN\.$', 'LANE', regex=True)
                      .str.replace(r'PL$|PL\.$', 'PLACE', regex=True)
                      .str.replace(r'TER$|TER\.$', 'TERRACE', regex=True)
                      .str.replace(r'HWY$|HWY\.$', 'HIGHWAY', regex=True)
                      .str.replace(r'PKWY$|PKWY\.$', 'PARKWAY', regex=True)
                      .str.replace(r'CIR$|CIR\.$', 'CIRCLE', regex=True))
            db.insert(0, 'Property Location', db.pop('Property Location').str.replace(r'.*', NJTaxAssessment.clean_addresses, regex=True))

            db.astype({'Sq. Ft.': 'int64'})

            return db

    @staticmethod
    def clean_addresses(search_string):

        target_list = str(search_string.group()).split(' ')
        new_address_list = [i for i in target_list if i != '']

        return ' '.join(new_address_list)

    @staticmethod
    def creation_time_delta(file_name, time_stamp):
        """
        Function which checks the delta between the file creation time and the time stamp argument.
        Used in conjuction with essex_county_scrape to match the file with the respective municipality
        :param file_name:
        :param time_stamp:
        :return:
        """
        abs_path = os.path.abspath(file_name)
        file_creation_timestamp = os.path.getctime(abs_path)
        file_creation_time = time.ctime(file_creation_timestamp)  #File_creation_timestamp is returned in secs from UNIX Epoch
        dt_object = datetime.strptime(file_creation_time, "%a %b %d %H:%M:%S %Y")
        delta = time_stamp - dt_object

        return delta

    @classmethod
    def create_state_dictionary(cls, driver_var):

        state_dict = {}

        try:

            url = NJTaxAssessment.tax_assessment
            url1 = 'https://www.taxdatahub.com/6229fbf0ce4aef911f9de7bc/Essex%20County'

            driver_var.get(url)
            results = driver_var.page_source
            counties = NJTaxAssessment.find_counties(results)

            for j in counties.keys():
                if counties[j] == 'ESSEX':
                    state_dict['ESSEX'] = []

                    driver_var.get(url1)
                    WebDriverWait(driver_var, 10).until(EC.presence_of_element_located(
                        (By.XPATH, "//*[@id='side-bar']/div[1]/h3")))

                    results1 = driver_var.page_source
                    value_pattern = re.compile(r'\d{4}')
                    soup = BeautifulSoup(results1, 'html.parser')
                    target = soup.find('div', {"id": "town-filter-div"})
                    target_contents = target.find_all('input', {'id': value_pattern})

                    for i in target_contents:
                        strip_list = [' Township', ' Borough', ' City']
                        for ending in strip_list:
                            if ending not in i['value']:
                                continue
                            else:
                                city = i['value'].rstrip(ending)
                                state_dict['ESSEX'].append(city.upper())

                else:
                    state_dict[counties[j]] = []
                    county_option = WebDriverWait(driver_var, 5).until(
                        EC.presence_of_element_located(
                            (By.XPATH, "//*[@id='normdiv']/form/table[1]/tbody/tr[3]/td[2]/select/option["
                             + str(j) + "]")))  # Make sure the XPATH includes the variable j
                    county_option.click()

                    results1 = driver_var.page_source
                    value_pattern = re.compile(r'\d{4}')
                    soup = BeautifulSoup(results1, 'html.parser')
                    target = soup.find('select', {"name": "district"})
                    target_contents = target.find_all('option', {'value': value_pattern})

                    for i in target_contents:
                        if i.get_text() == 'ALL':
                            break
                        else:
                            state_dict[counties[j]].append(i.get_text())

        except Exception as E:
            print(f'{E}')

        else:
            cls.state_county_dict = state_dict
            with shelve.open('NJ Tax Assessment Dictionary', writeback=True) as saved_data_file:
                saved_data_file['State Dictionary'] = cls.state_county_dict

    @staticmethod
    def database_check(county, city, logger):

        base_path = 'F:\\Real Estate Investing\\JQH Holding Company LLC\\Property Data'
        if county == 'ESSEX':
            strip_list = [' Township', ' Borough', ' City']
            for ending in strip_list:
                if ending not in city:
                    continue
                else:
                    city = city.rstrip(ending)
                    target_path = os.path.join(base_path, county, city.upper())
        else:
            target_path = os.path.join(base_path, county, city)

        if os.path.exists(target_path):
            latest_file = os.listdir(target_path)[-1]
            if str(datetime.today().year) in str(latest_file):
                logger.info(f'Database for {city} already exists')

                return 'File Already Exists'

        else:
            return "File Doesn't Exist"

    @staticmethod
    def essex_county_scrape(driver_var, city_name=None, **kwargs):
        """

        :param driver_var:
        :param city_name:
        :param kwargs:
        :return:
        """
        logger = kwargs['logger']
        f_handler = kwargs['f_handler']
        c_handler = kwargs['c_handler']

        proxies_dict = UniversalFunctions.format_proxies('F:\\Python 2.0\\Projects\\Real Life Projects\\Universal_Functions\\Proxies\\11-25-2023 proxy-list.txt')

        started_threads = []

        url = 'https://www.taxdatahub.com/6229fbf0ce4aef911f9de7bc/Essex%20County'

        driver_var.get(url)

        try:
            WebDriverWait(driver_var, 10).until(EC.presence_of_element_located(
                    (By.XPATH, "//*[@id='side-bar']/div[1]/h3")))

            results = driver_var.page_source
            cities = NJTaxAssessment.find_essex_cities(results)

            if city_name is None:
                # town_filter = WebDriverWait(driver_var, 5).until(EC.presence_of_element_located(
                #     (By.XPATH, "//*[@id='town-filter-el']/a/label")))
                # town_filter.click()
                for k in cities.keys():
                    try:

                        results = NJTaxAssessment.database_check('ESSEX', cities[k], logger)
                        if results == 'File Already Exists':
                            continue
                        else:
                            driver_var.proxy = {'https': UniversalFunctions.get_proxy(proxies_dict)}
                            driver_var.get(url + '?town=' + str(k))
                            WebDriverWait(driver_var, 10).until(EC.presence_of_element_located(
                                (By.XPATH, "//*[@id=" + k + "]")))  # Make sure the XPATH includes the variable k
                            # city_option.click()
                            # time.sleep(2)
                            download_link = WebDriverWait(driver_var, 10).until(EC.presence_of_element_located(
                                                (By.XPATH, "//a[normalize-space()='Download Excel']")))
                            time.sleep(2)
                            download_link.click()
                            time.sleep(2)
                            accept_disclaimer = WebDriverWait(driver_var, 10).until(EC.presence_of_element_located(
                                                (By.XPATH,  "//*[@id='notice-modal-download']/div/div/div[3]/button")))
                            accept_disclaimer.click()
                            # time.sleep(1.5)
                            waiting_for_download = WebDriverWait(driver_var, 30).until(EC.visibility_of_element_located(
                                                (By.XPATH, "//*[@id='search-bar']/div[2]/div[3]/div/button/span")))
                            while waiting_for_download:
                                download_done = WebDriverWait(driver_var, 90).until(EC.visibility_of_element_located(
                                                (By.XPATH, "//a[normalize-space()='Download Excel']")))
                                if not download_done:
                                    continue
                                else:
                                    NJTaxAssessment.unzip_and_extract('ESSEX', cities[k], time_stamp=datetime.now())
                                    logger.info(f'The download for {cities[k]} has finished...')
                                    time.sleep(1)
                                    # clear_filter = WebDriverWait(driver_var, 10).until(EC.presence_of_element_located(
                                    #     (By.XPATH, "//*[@id='town-filter-div']/label[1]")))
                                    # clear_filter.click()
                                    # break
                    except ElementNotInteractableException as ENI:
                        logger.exception(f'{"".join(format_tb(ENI.__traceback__))}')
                    except WebDriverException as WDE:
                        logger.exception(f'{"".join(format_tb(WDE.__traceback__))}')

            elif city_name is not None:
                pass

        except ElementNotVisibleException as ENV:  # Make more specific exception handling blocks later
            logger.exception(f'{ENV}')

        except ElementNotInteractableException as ENI:
            logger.exception(f'{ENI}')

        except ElementNotSelectableException as ENS:
            logger.exception(f'{ENS}')

        except InvalidArgumentException as IAE:
            logger.exception(f'{IAE}')

        except NoSuchAttributeException as NSAE:
            logger.exception(f'{NSAE}')

        except NoSuchDriverException as NSDE:
            logger.exception(f'{NSDE}')

        except NoSuchElementException as NSEE:
            logger.exception(f'{NSEE}')

        except WebDriverException as WDE:
            logger.exception(f'{WDE}')

        else:

            for threadobj in started_threads:
                threadobj.join()

            logger.info(f'All databases for ESSEX County have been downloaded')
            logger.removeHandler(f_handler)
            logger.removeHandler(c_handler)
            logging.shutdown()

    # @classmethod
    # def find_county(cls, city):
    #
    #     return cls.state_county_dict.loc[city, 'County']

    # @staticmethod
    # def find_city(county):
    #
    #     return NJTaxAssessment.state_county_dict.loc[city, 'County']

    @staticmethod
    def find_counties(page_source):
        """

        :param page_source:
        :return:
        """
        # Find the counties on the NJ Tax Assessment page
        value_pattern = re.compile(r'\d{4}')
        soup = BeautifulSoup(page_source, 'html.parser')
        target = soup.find('select', {"name": "select_cc"})
        target_contents = target.find_all('option', {'value': value_pattern})
        counties = {}

        for idx, i in enumerate(target_contents):
            counties[idx + 1] = i.get_text()

        keys = list(counties.keys())

        counties[keys[-1] + 1] = "ESSEX"  # Manually add ESSEX county to the dictionary because it's not in page source

        # NJTaxAssessment.state_county_dictionary(counties)

        return counties

    @staticmethod
    def find_cities(page_source):
        """

        :param page_source:
        :return:
        """
        # Find the counties on the NJ Tax Assessment page
        value_pattern = re.compile(r'\d{4}')
        soup = BeautifulSoup(page_source, 'html.parser')
        target = soup.find('select', {"name": "district"})
        target_contents = target.find_all('option', {'value': value_pattern})
        cities = {}

        for idx, i in enumerate(target_contents):
            if i.get_text() == 'ALL':
                break
            else:
                cities[idx + 1] = i.get_text()

        return cities

    @staticmethod
    def find_essex_cities(page_source):
        """

        :param page_source:
        :return:
        """
        # Find the counties on the NJ Tax Assessment page
        value_pattern = re.compile(r'\d{4}')
        soup = BeautifulSoup(page_source, 'html.parser')
        target = soup.find('div', {"id": "town-filter-div"})
        target_contents = target.find_all('input', {'id': value_pattern})
        cities = {}

        for i in target_contents:
            cities[i['id']] = i['value']

        return cities

    @staticmethod
    def find_zipfile(page_source):
        """

        :param page_source:
        :return:
        """
        try:
            # Find the zipfile for the NJ Tax Assessment download
            download_pattern = re.compile(r'(../download/(.*.zip))')
            soup = BeautifulSoup(page_source, 'html.parser')
            target = soup.find('a', {'href': download_pattern.search(str(soup)).group(0)})
            temp_name = download_pattern.search(str(target)).group(1)

        except Exception as e:
            print(f'{e}')

        return temp_name

    @classmethod
    def load_state_dictionary(cls):

        save_path = 'F:\\Python 2.0\\Projects\\Real Life Projects\\Real Estate Analysis\\NJ Tax Assessment Dictionary.dat'
        if os.path.exists(save_path):
            try:
                with shelve.open('NJ Tax Assessment Dictionary', writeback=True) as saved_data_file:
                    if saved_data_file['State Dictionary']:
                        cls.state_county_dict = saved_data_file['State Dictionary']
            except KeyError:
                print('A previously created State Dictionary does not exist. '
                      'Please execute NJTaxAssessment.create_state_dictionary to create one')

    @staticmethod
    @logger_decorator
    def nj_databases(driver_var, county_name=None, city_name=None, **kwargs):
        """

        :param driver_var:
        :param county_name:
        :param city_name:
        :param kwargs:
        :return:
        """
        # Method which can download all the ownership information for
        # existing homes in NJ off of the NJ Tax Assessment website

        try:
            # assert (city and county is None) or (city and county is not None)
            url = NJTaxAssessment.tax_assessment
            driver_var.get(url)

            results = driver_var.page_source
            counties = NJTaxAssessment.find_counties(results)

            if (county_name and city_name) is None:
                for j in counties.keys():
                    if counties[j] == 'ESSEX':
                        NJTaxAssessment.essex_county_scrape(driver_var, **kwargs)
                    else:
                        NJTaxAssessment.all_county_scrape(j, counties, driver_var, **kwargs)
            elif county_name is not None and city_name is None:
                pass

            elif county_name and city_name is not None:
                pass

        except Exception as E:
            print(f'{E}')

    @classmethod
    def state_county_dictionary(cls):
        """

        :return:
        """
        return cls.state_county_dict

    @staticmethod
    def stream_zipfile(base_url, download_param, session_var):

        previous_dir = os.getcwd()
        os.chdir('C:\\Users\\Omar\\Desktop\\Selenium Temp Folder')
        new_filename = download_param.split('/')[2]
        full_url = "".join([base_url, download_param.lstrip('../')])

        with session_var.get(full_url, stream= True) as reader, open(new_filename, 'wb') as writer:
            for chunk in reader.iter_content(chunk_size=1000000):
                writer.write(chunk)

        os.chdir(previous_dir)

    def tax_lien_foreclosure(self, county=None, city=None):
        """

        :param county:
        :param city:
        :return:
        """
        # Creates a database of all the city owned and vacant lots
        # and properties for future pursuit. Uses the nj_database file

        if city and county is None:
            db = self.database

        else:
            db = NJTaxAssessment.city_database(county, city)

        if 'EPL_Code' in db.columns:  # Only used for municipalities in Essex County
            tax_lien_db = db[db['EPL_Code'] == '0401047']
        else:
            tax_lien_db = db[(db['EPL Own'] == 4) & (db['EPL Use'] == 1) & (db['EPL Desc'] == 47)]

        return tax_lien_db

    @staticmethod
    def unzip_and_extract(county, city, temp_file_name=None, time_stamp=None):
        """
        Will find the recently downloaded zip file of the city
        for which all the property information is located. This function will accept the
        temporary file name and city as arguments and rename the file with the respective
        city in the name and store it in the specific directory under that county
        :param county:
        :param city:
        :param temp_file_name:
        :param time_stamp:
        :return:
        """

        previous_dir = os.getcwd()
        path = 'C:\\Users\\Omar\\Desktop\\Selenium Temp Folder'
        base_path = 'F:\\Real Estate Investing\\JQH Holding Company LLC\\Property Data'
        os.chdir(path)
        filenames = os.listdir(path)

        try:
            for file in filenames:

                if county == 'ESSEX':
                    strip_list = [' Township', ' Borough', ' City']
                    for ending in strip_list:
                        if ending not in city:
                            continue
                        else:
                            city = city.rstrip(ending)
                            target_path = os.path.join(base_path, county, city.upper())
                else:
                    target_path = os.path.join(base_path, county, city)

                if temp_file_name is not None:
                    if temp_file_name != file:
                        continue

                    elif temp_file_name == file:
                        extract_file = ZipFile(os.path.abspath(file))
                        target_file = temp_file_name.rstrip('.zip') + '.csv'
                        extract_file.extract(target_file)
                        extract_file.close()

                        if os.path.exists(target_path):
                            time.sleep(0.5)
                            shutil.move(os.path.abspath(target_file), os.path.join(target_path, city + ' '
                                                        + 'Database' + ' ' + str(datetime.today().date()) + '.csv'))
                            send2trash.send2trash(file)
                        else:
                            os.makedirs(target_path)
                            time.sleep(0.5)
                            shutil.move(os.path.abspath(target_file), os.path.join(target_path, city + ' '
                                                            + 'Database' + ' ' + str(datetime.today().date()) + '.csv'))
                            send2trash.send2trash(file)
                elif time_stamp is not None:
                    difference = NJTaxAssessment.creation_time_delta(file, time_stamp)
                    if difference > timedelta(seconds=10):
                        continue
                    else:
                        target_file = file + '.xlsx'
                        if os.path.exists(target_path):
                            time.sleep(0.5)
                            shutil.move(os.path.abspath(target_file), os.path.join(target_path, city + ' '
                                                            + 'Database' + ' ' + str(datetime.today().date()) + '.xlsx'))
                        else:
                            os.makedirs(target_path)
                            time.sleep(0.5)
                            shutil.move(os.path.abspath(target_file), os.path.join(target_path, city + ' '
                                                                + 'Database' + ' ' + str(datetime.today().date()) + '.xlsx'))
                # else:
                #     if temp_file_name is not None:
                #         raise IndexError(f"File Does Not Exist: {temp_file_name}")
                #     else:
                #         raise IndexError(f"File Does Not Exist", county, city)
        except IndexError as IE:
            print(f'{IE} ----> {city}')

        except Exception as E:
            print(f'{E}')

        else:
            os.chdir(previous_dir)

    def vacant_land(self, county=None, city=None):
        """

        :param county:
        :param city:
        :return:
        """
        # Creates a database of all the city owned and vacant lots
        # and properties for future pursuit. Uses the nj_database file

        if city and county is None:
            db = self.database

        else:
            db = NJTaxAssessment.city_database(county, city)

        if 'EPL_Code' in db.columns:  # Only used for municipalities in Essex County
            vacant_land_db = db[db['PropertyClassCode'] == 1]
        else:
            vacant_land_db = db[db['Property Class'] == 1]

        return vacant_land_db

    def main(self):
        try:
            # uv = UniversalFunctions
            proxy_dict = UniversalFunctions.format_proxies('F:\\Python 2.0\\Projects\\Real Life Projects\\Universal_Functions\\Proxies\\11-25-2023 proxy-list.txt')
            save_location1 = 'C:\\Users\\jibreel.q.hameed\\Desktop\\Selenium Temp Folder'
            save_location2 = 'C:\\Users\\Omar\\Desktop\\Selenium Temp Folder'  # May need to be changed
            options = Options()
            sw_options = {'proxy': {'https': UniversalFunctions.get_proxy(proxy_dict)}}
            # Change this directory to the new one: ('C:\\Users\\Omar\\Desktop\\Python Temp Folder')
            s = {"savefile.default_directory": save_location2,
                 "download.default_directory": save_location2,
                 "download.prompt_for_download": False}
            # options.add_experimental_option("detach", True)
            options.add_experimental_option("prefs", s)
            options.add_argument("--start-maximized")
            options.add_argument("--headless=new")

            driver = sw_webdriver.Edge(service=Service(), options=options, seleniumwire_options=sw_options)

            # NJTaxAssessment.create_state_dictionary(driver)
            # NJTaxAssessment.nj_databases(driver)

        except Exception as e:
            print(e)


if __name__ == '__main__':

    obj = NJTaxAssessment()
    obj.main()

    # print(os.getcwd())
    # print(sys.path)