import os
import shutil
import sys
import pymysql
from peewee import *
import yaml

outputDir = "output"
# Clean the output dir & create a new
shutil.rmtree(outputDir, True)
os.mkdir(outputDir)

# Connect to the MySQL database
db = pymysql.connect(
    host='localhost',
    db='spip',
    user='spip',
    password='password',
)

# Query the database to retrieve all data
cursor = db.cursor()
cursor.execute('SELECT * FROM spip_articles ORDER BY date DESC')

# Loop through the results and format data into Markdown files
# Columns:
#   2: titre
#   7: texte
#   9: date

if len(sys.argv) > 1:
    if int(sys.argv[1]) > 0:
        fetch = cursor.fetchmany(int(sys.argv[1]) + 1)
    else:
        fetch = cursor.fetchall()
else:
    fetch = cursor.fetchmany(3 + 1)

for row in fetch:
    frontmatter = {'title': row[2], 'date': row[9]}
    content = f'---\n{yaml.dump(frontmatter)}---\n{row[2]}\n\n{row[7]}'
    path = f'{outputDir}/{row[2]}.md'
    with open(path, 'w') as f:
        f.write(content)

# Close the database connection
db.close()