app.py

import tabula
import math
import json
import os
import random
from covid19 import Nhx
from tabulate import tabulate
import platform
import matplotlib as mpl
import matplotlib.pyplot as plt
import numpy as np

OS = platform.system()
scriptDir = os.path.dirname(os.path.abspath(__file__))

def getTable(src):
    percobaan = []
    df = tabula.read_pdf(src, output_format="json", pages='all')
    rows = df[0]['data']
    for row in rows:
        data = row
        p = []
        for datum in data:
            p.append(datum['text'])

        percobaan.append(p)

    return percobaan

def completeTable(table):
    for row in table[1:]:
        for i, data in enumerate(row[1:]):
            if '?' in data:
                row[i+1] = calculateAnswer(i, row[1:])

def calculateAnswer(question, data):
    ans = 'None'
    if question == 5:
        Nh, h = Nhx(float(data[0]), float(data[1]), float(data[2]), float(data[4]))
        ans = math.floor(Nh)
    elif question == 3:
        ans = '1 + {}'.format(float(data[1])*float(data[2]))
    return '{}'.format(ans)

def createGraph(data):
    if OS == 'Windows':
        if not os.path.exists(scriptDir+'\Grafik'):
            os.makedirs(scriptDir+'\Grafik')
    else:
        if not os.path.exists(scriptDir+'/Grafik/'):
            os.makedirs(scriptDir+'/Grafik/')
        

    info = 'Percobaan Ke-{}'.format(data[0])
    filename = 'pcbk{}'.format(data[0])
    N = float(data[1])
    R = float(data[2])
    P = float(data[3])
    mx = int(data[5])

    x = []
    y = []
    for i in range(mx):
        Nh, h = Nhx(N, R, P, i)
        y.append(Nh)
        x.append(h+1)

    fig, ax = plt.subplots()
    ax.plot(x, y)

    plt.xticks(x)

    ax.set(xlabel='Hari', ylabel='Jumlah Kasus',
       title=info)
    ax.grid()

    if OS == 'Windows':
        fig.savefig('Grafik\{}.png'.format(filename))
    else:
        fig.savefig('Grafik/{}.png'.format(filename))

table = getTable('source.pdf')
os.system('cls' if OS == 'Windows' else 'clear')
completeTable(table)

style=[
    "plain",
    "simple",
    "github",
    "grid",
    "fancy_grid",
    "pipe",
    "orgtbl",
    "jira",
    "presto",
    "pretty",
    "psql",
    "rst",
    "mediawiki",
    "moinmoin",
    "youtrack",
    "latex",
    "latex_booktabs",
    "textile",
]
graphStyle = plt.style.available

print(tabulate(table, headers="firstrow", tablefmt=style[random.randint(0, len(style))]))

mpl.style.use(graphStyle[random.randint(0, len(graphStyle))])

for row in table[1:]:
    createGraph(row)