libreCatastro/main.py

38 lines
1.6 KiB
Python

import sys
import argparse
from src.librecatastro.scrapping.format.scrapper_html import ScrapperHTML
from src.librecatastro.scrapping.format.scrapper_xml import ScrapperXML
from src.librecatastro.scrapping.source.coordinates_input import CoordinatesInput
from src.librecatastro.scrapping.source.provinces_input import ProvincesInput
from src.settings import config
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Runs the Cadastro Parser')
parser.add_argument('--coords', action='store_true', dest='coords', default=False)
parser.add_argument('--filenames', action='store', nargs='+', dest='filenames', default=[])
parser.add_argument('--provinces', action='store', nargs='+', dest='provinces', default=[])
parser.add_argument('--sleep', action='store', dest='sleep', type=int, default=5)
parser.add_argument('--html', dest='html', default=False, action='store_true')
parser.add_argument('--scale', action='store', dest='scale', type=int, default=10000)
parser.add_argument('--pictures', action='store_true', dest='pictures', default='False')
args = parser.parse_args(sys.argv[1:])
if args.sleep:
config['sleep_time'] = args.sleep
if args.scale:
config['scale'] = args.scale
scrapper = ScrapperHTML if args.html else ScrapperXML
filenames = args.filenames
pictures = args.pictures
provinces = args.provinces
if args.coords:
CoordinatesInput.scrap_coordinates(scrapper, filenames, pictures)
else:
ProvincesInput.scrap_provinces(scrapper, provinces, pictures)