HomeHarvest/homeharvest/utils.py

49 lines
1.3 KiB
Python
Raw Normal View History

import re
2023-09-18 15:42:16 -07:00
def parse_address_two(street_address: str) -> tuple:
if not street_address:
return street_address, None
apt_match = re.search(
2023-09-18 15:57:15 -07:00
r"(APT\s*[\dA-Z]+|#[\dA-Z]+|UNIT\s*[\dA-Z]+|LOT\s*[\dA-Z]+|SUITE\s*[\dA-Z]+)$",
2023-09-18 15:42:16 -07:00
street_address,
re.I,
)
if apt_match:
apt_str = apt_match.group().strip()
2023-09-18 15:57:15 -07:00
cleaned_apt_str = re.sub(
r"(APT\s*|UNIT\s*|LOT\s*|SUITE\s*)", "#", apt_str, flags=re.I
)
2023-09-18 15:42:16 -07:00
2023-09-18 15:57:15 -07:00
main_address = street_address.replace(apt_str, "").strip()
2023-09-18 15:42:16 -07:00
return main_address, cleaned_apt_str
else:
return street_address, None
def parse_unit(street_address: str):
2023-09-18 15:04:34 -07:00
if not street_address:
return None
2023-09-18 15:04:54 -07:00
apt_match = re.search(
r"(APT\s*[\dA-Z]+|#[\dA-Z]+|UNIT\s*[\dA-Z]+|LOT\s*[\dA-Z]+)$",
street_address,
re.I,
)
2023-09-18 15:04:34 -07:00
if apt_match:
apt_str = apt_match.group().strip()
apt_str = re.sub(r"(APT\s*|UNIT\s*|LOT\s*)", "#", apt_str, flags=re.I)
return apt_str
else:
return None
2023-09-18 15:04:54 -07:00
2023-09-18 15:04:34 -07:00
if __name__ == "__main__":
2023-09-18 15:42:16 -07:00
print(parse_address_two("4303 E Cactus Rd Apt 126"))
2023-09-18 15:04:34 -07:00
print(parse_address_two("1234 Elm Street apt 2B"))
print(parse_address_two("1234 Elm Street UNIT 3A"))
print(parse_address_two("1234 Elm Street unit 3A"))
2023-09-18 15:57:15 -07:00
print(parse_address_two("1234 Elm Street SuIte 3A"))