48 lines
1.0 KiB
Python
48 lines
1.0 KiB
Python
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()
|