proca_examples.py

#! python3
#
# Here is some examples of using proca classes
# and working with production calendar data stored in JSON files


import proca
import json


# Returns dictionary loaded from JSON file
def readJSON(fileName):
    with open(fileName) as fileJSON:
        data = json.load(fileJSON)
    return data


# Construct object 2023 year production calendar
# from source JSON file
year = proca.ProcaYear(readJSON('sjob_min_2023.json'))

# Make 1st quarter months of the year
months = year(1, 2, 3, 4)

# Get January by alias 'jan'
jan = months.jan

# Print January days table
print(jan)

# Show January info using month fields
print(f'{jan.name}:',
      f'&#1088;&#1072;&#1073;&#1086;&#1095;&#1080;&#1093; &#1076;&#1085;: {jan.twork}',
      f'&#1074;&#1099;&#1093;&#1086;&#1076;&#1085;&#1099;&#1093; &#1076;&#1085;: {jan.trest}',
      f'&#1074;&#1089;&#1077;&#1075;&#1086; &#1076;&#1085;: {jan.tdays}',
      sep='\n')

# Get 7th day of January by alias 'd7'
d7 = jan.days.d7

# Print day info table
print(d7)

# Show day info using day fields
print(f'&#1076;&#1072;&#1090;&#1072;: {d7.dnum:02d}.{d7.mnum:02d}.{d7.ynum}',
      f'&#1076;&#1077;&#1085;&#1100; &#1085;&#1077;&#1076;&#1077;&#1083;&#1080;: {d7.name}',
      f'&#1074;&#1099;&#1093;&#1086;&#1076;&#1085;&#1086;&#1081; &#1076;&#1077;&#1085;&#1100;: {"&#1044;&#1072;" if d7.rest == 1 else "&#1053;&#1077;&#1090;"}',
      f'&#1087;&#1088;&#1072;&#1079;&#1076;&#1085;&#1080;&#1082;: {d7.ttip if d7.dtype == 3 else "&#1053;&#1077;&#1090;"}',
      sep='\n')

# Make 12 months (it's default if not args passed)
months = year()

# Table header
stick = ' | '
fields_width = {'mnum': 2, 'name': 8, 'tdays': 4, 'twork': 4, 'trest': 4}
fields_ru = {'mnum': '#', 'name': '&#1084;&#1077;&#1089;.', 'tdays': '&#1076;&#1085;.',
             'twork': '&#1088;&#1072;&#1073;.', 'trest': '&#1074;&#1099;&#1093;.'}

# Show table header
print(*[fields_ru[f].ljust(w) for f, w in fields_width.items()], sep=stick)

# Show top line
print('-' * (sum(fields_width.values()) + len(stick)*4))

# Show months info table
for m in months:
    row = []
    for f, w in fields_width.items():
        if f == 'mnum':
            # 2 pos digit with leading zero
            cell = f'{getattr(m, f):02d}'.rjust(w)
        else:
            cell = str(getattr(m, f)).ljust(w)
        row.append(cell)
    print(*row, sep=stick)

# Show bottom line
print('-' * (sum(fields_width.values()) + len(stick)*4))

# Calculate totals by 12 months
fields_total = ('tdays', 'twork', 'trest')
totals = {f: sum(getattr(m, f, 0) for m in months) for f in fields_total}
# Show table totals
print(*[str(totals[f]).ljust(w)
        if f in fields_total
        else ''.ljust(w)
        for f, w in fields_width.items()], sep=' | ')