# 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)