import sys, csv, datetime

fileName = sys.argv[1] # e.g. stmf.csv
countryCode = sys.argv[2] # e.g. USA
figureFieldName = sys.argv[3] # e.g. D0_14 (deaths for 0-14y), DTotal (total deaths)
smoothingCount = int(sys.argv[4]) if len(sys.argv) >= 4 + 1 else 1 # e.g. 3

data = []
with open(fileName) as file1:
  file1.readline() # Two intro lines without data
  file1.readline()
  for line in csv.DictReader(file1):
    if line and line["CountryCode"] == countryCode and line["Sex"] == "b":
      deaths = int(float(line[figureFieldName]))
      date = line["Year"] + " " + line["Week"] + " 0"
      date = datetime.datetime.strptime(date, "%Y %U %w")
      data.append( (date, deaths) )

data.pop() # Drop last two weeks for too big registration delay effect
data.pop()

def movingAverage(values, itemCount):
  average = 0
  averages = []
  idx = -1
  for val in values:
    idx += 1
    average += val - values[idx - itemCount] if idx >= itemCount else val
    if idx >= itemCount - 1:
      averages.append(average / float(itemCount))
    else:
      averages.append(None)
  return averages

values = [v for k, v in data]
if smoothingCount > 1:
  values = movingAverage(values, smoothingCount)

sys.stdout.write("|x = " + ", ".join([k.strftime("%Y-%m-%d") for k, v in data]) + "\n")
sys.stdout.write("|y = " + ", ".join(["%.0f" % v if v is not None else "" for v in values]) + "\n")

import sys, csv, datetime

fileName = sys.argv[1] # e.g. stmf.csv
countryCode = sys.argv[2] # e.g. USA
figureFieldName = sys.argv[3] # e.g. D0_14 (deaths for 0-14y), DTotal (total deaths)

data = []
file1 = open(fileName)
file1.readline() # Two intro lines without data
file1.readline()
for line in csv.DictReader(file1):
  if line and line["CountryCode"] == countryCode and line["Sex"] == "b":
    deaths = int(float(line[figureFieldName]))
    data.append( (int(line["Year"]), int(line["Week"]), deaths) ) 

data.pop() # Drop last two weeks for too big registration delay effect
data.pop()

year2020Weeks = [week for year, week, deaths in data if year == 2020]
maxWeek = year2020Weeks[-1]
years = sorted(list({year for year, week, deaths in data}))

deathsUpToMaxWeek = []
for year in years:
  deathsUpToMaxWeek1 = 0
  for year1, week, deaths in data:
    if year1 == year and week <= maxWeek:
      deathsUpToMaxWeek1 += deaths
  deathsUpToMaxWeek.append(deathsUpToMaxWeek1)
       
yearsOut = ", ".join([str(year) for year in years])
deathsOut = ", ".join([str(deaths) for deaths in deathsUpToMaxWeek])
sys.stdout.write("Last week in 2020: %i\n" % maxWeek)
sys.stdout.write("|x = " + yearsOut + "\n")
sys.stdout.write("|y = " + deathsOut + "\n")

import sys, csv, datetime

# Plot deaths per season: week 40 of year before to week x of the year

fileName = sys.argv[1] # e.g. stmf.csv
countryCode = sys.argv[2] # e.g. USA
figureFieldName = sys.argv[3] # e.g. D0_14 (deaths for 0-14y), DTotal (total deaths)
seasonStartWeek = 40

data = []
file1 = open(fileName)
file1.readline() # Two intro lines without data
file1.readline()
for line in csv.DictReader(file1):
  if line and line["CountryCode"] == countryCode and line["Sex"] == "b":
    deaths = int(float(line[figureFieldName]))
    data.append( (int(line["Year"]), int(line["Week"]), deaths) ) 

data.pop() # Drop last two weeks for too big registration delay effect
data.pop()

year2020Weeks = [week for year, week, deaths in data if year == 2020]
maxWeek = year2020Weeks[-1]
years = sorted(list({year for year, week, deaths in data}))

deathsInSeason = []
for year in years[1:]:
  deathsInSeason1 = 0
  for year1, week, deaths in data:
    if year1 == year and week <= maxWeek:
      deathsInSeason1 += deaths
    if year1 == (year - 1) and week >= seasonStartWeek:
      deathsInSeason1 += deaths      
  deathsInSeason.append(deathsInSeason1)
       
yearsOut = ", ".join([str(year) for year in years[1:]])
deathsOut = ", ".join([str(deaths) for deaths in deathsInSeason])
write = sys.stdout.write
write("Last week in 2020: %i\n" % maxWeek)
write("All-cause deaths in weeks %i+ year before and weeks 1-%i of the year, year by year:\n" % (seasonStartWeek, maxWeek))
write("|x = " + yearsOut + "\n")
write("|y = " + deathsOut + "\n")

import sys, csv, datetime

fileName = sys.argv[1] # e.g. "Weekly_excess_deaths_Full_Data_data.csv"
data = []
for line in csv.reader(open(fileName)):
  if "Observed Number" not in line[0]:
    date =  datetime.datetime.strptime(line[2], "%B %d, %Y")
    deaths = line[0].replace(",", "")
    data.append( (date, deaths) )

data.sort(key=lambda x: x[0])

datesOut = ", ".join([k.strftime("%Y-%m-%d") for k, v in data])
deathsOut = ", ".join([v for k, v in data])
sys.stdout.write("|x =" + datesOut + "\n")
sys.stdout.write("|y =" + deathsOut + "\n")