90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
from flask import Flask, render_template
|
|
from datetime import datetime
|
|
import os
|
|
import re
|
|
|
|
app = Flask(__name__)
|
|
|
|
def get_dates_from_images(image_directory):
|
|
# Regular expression to match files with a date format YYYY-MM-DD
|
|
date_pattern = re.compile(r'(\d{4}-\d{2}-\d{2})')
|
|
|
|
# List to store the dates
|
|
dates = []
|
|
|
|
# Iterate over each file in the images directory
|
|
for filename in os.listdir(image_directory):
|
|
# Search for the pattern in the filename
|
|
match = date_pattern.search(filename)
|
|
if match:
|
|
# Extract the date from the filename
|
|
try:
|
|
date = datetime.strptime(match.group(), '%Y-%m-%d')
|
|
dates.append(date)
|
|
except ValueError:
|
|
# If the date format is incorrect, ignore the file
|
|
continue
|
|
|
|
# Return the sorted list of dates
|
|
return sorted(dates)
|
|
|
|
def get_image_and_text_for_date(date, image_directory, text_directory):
|
|
formatted_date = date.strftime('%Y-%m-%d')
|
|
image_file = None
|
|
text_content = None
|
|
|
|
# Check for image file
|
|
for ext in ['jpg', 'jpeg', 'png']:
|
|
# Check using full path, but store relative path for HTML
|
|
image_full_path = os.path.join(image_directory, f"{formatted_date}.{ext}")
|
|
if os.path.isfile(image_full_path):
|
|
# Store the relative path from within the 'static' directory
|
|
image_file = f"images/{formatted_date}.{ext}"
|
|
break
|
|
|
|
# Check for text file
|
|
text_path = os.path.join(text_directory, f"{formatted_date}.txt")
|
|
if os.path.isfile(text_path):
|
|
with open(text_path, 'r') as file:
|
|
text_content = file.read()
|
|
|
|
return image_file, text_content
|
|
|
|
|
|
@app.route('/')
|
|
def calendar():
|
|
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
image_directory = os.path.join(BASE_DIR, 'static', 'images')
|
|
text_directory = os.path.join(BASE_DIR, 'texts')
|
|
|
|
# Get sorted dates from image files
|
|
dates = get_dates_from_images(image_directory)
|
|
if not dates:
|
|
return "No images found with the correct date format in the name.", 404
|
|
|
|
# Generate data for each date
|
|
pages_data = []
|
|
days_data = []
|
|
for date in dates:
|
|
image_file, text_content = get_image_and_text_for_date(date, image_directory, text_directory)
|
|
days_data.append({
|
|
'date': date,
|
|
'image_file': image_file,
|
|
'text_content': text_content
|
|
})
|
|
|
|
# Check if we have collected data for 4 days, then start a new page
|
|
if len(days_data) == 4:
|
|
pages_data.append(days_data)
|
|
days_data = []
|
|
|
|
# Add any remaining days to the last page
|
|
if days_data:
|
|
pages_data.append(days_data)
|
|
|
|
return render_template('index.html', pages=pages_data)
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True)
|