markup: add season info from pdf

This commit is contained in:
2019-10-18 16:39:34 -07:00
parent f819ac65b5
commit 2b498fd2e3
2 changed files with 8 additions and 5 deletions

View File

@ -11,17 +11,20 @@ from .utils import pdf_rect, ensure_dir, set_file_perms, WORKDIR
def make_product_box(obj, pagenum, mediabox): def make_product_box(obj, pagenum, mediabox):
name = obj['ProCatName'].decode()
material = obj['ProCatMaterialNumber'].decode()
color = obj['ProCatColor'].decode()
gender = obj['ProCatGender'].decode()
rect = obj['Rect'] rect = obj['Rect']
if rect: if rect:
name = obj['ProCatName'].decode() if 'ProCatName' in obj else ''
material = obj['ProCatMaterialNumber'].decode() if 'ProCatMaterialNumber' in obj else ''
color = obj['ProCatColor'].decode() if 'ProCatColor' in obj else ''
gender = obj['ProCatGender'].decode() if 'ProCatGender' in obj else ''
season = obj['ProCatSeason'].decode() if 'ProCatSeason' in obj else ''
return { 'material': material, return { 'material': material,
'name': name, 'name': name,
'color': color, 'color': color,
'gender': gender, 'gender': gender,
'season': season,
'rect': pdf_rect(rect, mediabox[3]), 'rect': pdf_rect(rect, mediabox[3]),
'page': pagenum } 'page': pagenum }
else: else:

View File

@ -30,7 +30,7 @@ def overlaps(r1, r2, threshold):
SU = SA + SB - SI SU = SA + SB - SI
overlap = float(SI) / float(SU) overlap = float(SI) / float(SU)
print('overlap: {}'.format(int(overlap * 100))) #print('overlap: {}%'.format(int(overlap * 100)))
return overlap > threshold return overlap > threshold