import json
import requests
from matplotlib import pyplot as plt
import datetime
import struct
import ctypes
import os

api_url = 'https://covid19.rthand.com/api/data'

response = requests.get(api_url)

print('sup')

# print(response.status_code)

raw_data = response._content

json_data = json.loads(raw_data)

# print(type(json_data))

indexes = []
positive_list = []
dead_list = []
daily_positive_list = []
hospitalized_list = []
i = 1
for x, d in enumerate(json_data):
    # print(d['positiveTestsToDate'])
    print(d)
    if d['positiveTestsToDate'] == None:
        continue
    if x == len(json_data) - 1 and d['positiveTestsToDate'] == 0:
        print('last one!! and breaking')
        break
    indexes.append(i)
    i = i + 1
    positive_list.append(d['positiveTestsToDate'])
    daily_positive_list.append(d['positiveTests'])
    state_per_treatment = d['statePerTreatment']

    if state_per_treatment['deceasedToDate'] is not None:
        dead_list.append(state_per_treatment['deceasedToDate'])
    else:
        dead_list.append(0)

    if state_per_treatment['deceasedToDate'] is not None:
        hospitalized_list.append(state_per_treatment['inHospital'])
    else:
        hospitalized_list.append(0)


plt.figure(figsize=(21, 9))
plt.plot(indexes, positive_list, c='orange')
plt.plot(indexes, dead_list, c='r')
plt.plot(indexes, hospitalized_list, c='violet')
# print(daily_positive_list)
plt.bar(indexes, daily_positive_list)


tfs = 8

for i, v in enumerate(daily_positive_list):
    plt.text(i + 1, v + 2, str(v), horizontalalignment='center', c='blue', fontsize=tfs)

for i, v in enumerate(positive_list):
    plt.text(i + 1, v + 10, str(v), horizontalalignment='center', c='orange', fontsize=tfs)

for i, v in enumerate(hospitalized_list):
    if v == 0: continue
    plt.text(i + 1, v + 10, str(v), horizontalalignment='center', c='violet', fontsize=tfs)

plt.legend(['Skupaj pozitivni', 'Skupaj mrtvi', 'Hospitalizirani', 'Dnevno okuženi'])
plt.xticks(range(1, len(positive_list) + 1))
plt.yticks(range(0, max(positive_list), 50))
plt.ylabel('Št. ljudi')
plt.xlabel('Dnevi od začetka')

plt.grid()


dtobj = datetime.date.today()
plt.title('{0}.{1}.{2}'.format(dtobj.day, dtobj.month, dtobj.year))
fname = 'coronabg-{0}.png'.format(str(dtobj))

dirpath = os.getcwd()
print(dirpath)
img_path = '{0}\\img'.format(dirpath)

full_path = '{0}\\{1}'.format(img_path, fname)

if not os.path.exists(img_path):
    os.makedirs(img_path)

plt.savefig(full_path, dpi=600)
# plt.show()


PATH = full_path
SPI_SETDESKWALLPAPER = 20

ctypes.windll.user32.SystemParametersInfoW(SPI_SETDESKWALLPAPER, 0, PATH, 3)