import matplotlib.pyplot as plt
import matplotlib.dates as mdates
from datetime import datetime
from matplotlib.ticker import MultipleLocator

# vars
dt_format = '%Y-%m-%d'

# Initialize lists to store the dates and values
dates = []
values = []

# format of mylpale_data.txt is as follows
# 2001-11-01 1
# 2002-08-01 50
# 2003-04-01 100
# etc.

# Open the text file and read the data
with open('mylpale_data.txt', 'r') as file:
    for line in file:
        # Split the line into date and value
        date_str, value_str = line.strip().split()
        # Convert the date string to a datetime object
        date = datetime.strptime(date_str, dt_format)
        # Convert the value string to an integer
        value = int(value_str)
        # Append the date and value to their respective lists
        dates.append(date)
        values.append(value)

# Create a figure and set the size
fig, ax = plt.subplots(figsize=(12, 6))  # Adjust the width and height as needed

# Plot the data
ax.plot_date(dates, values, 'o-', markersize=5, linewidth=2, color='#0060ad')

# Set the title and labels
ax.set_title("Aantal Afrikaanse Wikipedia Artikels", fontsize=14)
ax.set_xlabel("Datum")
ax.set_ylabel("Aantal Artikels")

# Set the x-axis major ticks to years
ax.xaxis.set_major_locator(mdates.YearLocator())
ax.xaxis.set_major_formatter(mdates.DateFormatter('%Y'))

# Set the x-axis range
start_date = datetime.strptime('2001-06-01', dt_format)
end_date = datetime.strptime(f"{datetime.now().year + 2}-01-01", '%Y-%m-%d')  # Use current year + 2 as the end date
ax.set_xlim(start_date, end_date)

# Set y-axis range
ax.set_ylim(0, (values[-1] + 5000))

# Set the y-axis gridlines to every 5000
ax.yaxis.set_major_locator(MultipleLocator(5000))

# Remove the x-axis and y-axis ticks
ax.xaxis.set_tick_params(which='both', bottom=False, top=False, labelbottom=True)
ax.yaxis.set_tick_params(which='both', left=False, right=False, labelleft=True)

# Add a grid
ax.grid(True)

# Add padding to the plot
ax.margins(x=0.1, y=0.1)  # Adjust x and y padding as needed

# Save the plot as an SVG file
plt.savefig("mylpale_fat.svg", format='svg', dpi=1200)

print('Klaar')