diff options
| -rw-r--r-- | scrap/fincaraiz.py | 3 | ||||
| -rw-r--r-- | scrap/scrap.py | 11 |
2 files changed, 12 insertions, 2 deletions
diff --git a/scrap/fincaraiz.py b/scrap/fincaraiz.py index e8adcf0..354f407 100644 --- a/scrap/fincaraiz.py +++ b/scrap/fincaraiz.py @@ -87,7 +87,7 @@ JSON_DATA = { } } -def get_json(city: str, pages: int = 1) -> list: +def get_json(city: str, pages: int = 1, offer: list[str] = ["sell"]) -> list: """Get json data from fincaraiz.com.co Parameters @@ -129,6 +129,7 @@ def get_json(city: str, pages: int = 1) -> list: for x in requests.get("https://www.datos.gov.co/resource/xdk5-pm3f.json").json() if town_exist(x['municipio'])] + json_data["filter"]["offer"]["slug"] = offer if location: json_data["filter"]["locations"] = {"cities": {"slug": location } } else: diff --git a/scrap/scrap.py b/scrap/scrap.py index 50ef3d4..946a2a4 100644 --- a/scrap/scrap.py +++ b/scrap/scrap.py @@ -7,6 +7,9 @@ Options: -v --verbose Be more verbose -o FILE Redirect standard output to FILE. -p INT Number of pages to get [default: 1] + -r --rent Select housing rent offers. + -s --sell Select housing sell offers, if there is no -r option this + option is enabled by default. Examples: $ scrap Medellín -p 20 @@ -24,7 +27,13 @@ def main(): args = docopt(__doc__) city: str = args["<city>"] - options = {"pages": int(args["-p"])} + options = {"pages": int(args["-p"]), + "offer": ["sell"]} + + if args["-s"] and args["-r"]: + options["offer"].append("rent") + elif args["-r"] and not args["-s"]: + options["offer"] = ["rent"] json_data = get_json(city, **options) |