viktor/wrongmove
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixing floorplan detection and adding recalculation method

Browse source
This commit is contained in:
Kadir 2024-03-18 00:56:39 +00:00
parent 335adc0856
commit 4dea766a12
4 changed files with 21 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
crawler/9_recalculate_regex_squaremeter.py Normal file
View file

@ -0,0 +1,15 @@
# recalculate regex from sqm from already previously ocr'ed text
import json
from rec.floorplan import extract_total_sqm
from tqdm import tqdm
from data_access import Listing
for listing in tqdm(list(Listing.get_all_listings())):
with open(listing.path_floorplan_ocr_json()) as f:
floorplans = json.load(f)
for floorplan in floorplans:
floorplan['estimated_sqm'] = extract_total_sqm(floorplan['text'])
with open(listing.path_floorplan_ocr_json(), 'w') as f:
floorplans = json.dump(floorplans, f)