Add PDF renamer web app with ZIP support, styling, and logo
This commit is contained in:
nmemmert committed
1 parent
6f3a551b83
commit
3706314c93
4 files changed
+182
-82
No files matched your search
+157
@@ -0,0 +1,157 @@
|
||||
import os
|
||||
import tempfile
|
||||
import zipfile
|
||||
from flask import Flask, request, render_template_string, send_from_directory
|
||||
from werkzeug.utils import secure_filename
|
||||
from PyPDF2 import PdfReader
|
||||
|
||||
app = Flask(__name__)
|
||||
app.config['UPLOAD_FOLDER'] = tempfile.mkdtemp()
|
||||
app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024 # 100MB limit
|
||||
|
||||
ALLOWED_EXTENSIONS = {'pdf', 'zip'}
|
||||
|
||||
def allowed_file(filename):
|
||||
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
|
||||
|
||||
def extract_names_from_pdf(pdf_path):
|
||||
try:
|
||||
reader = PdfReader(pdf_path)
|
||||
fields = reader.get_fields()
|
||||
last_name = fields.get('Last Name', {}).get('/V', '')
|
||||
first_name = fields.get('First Name', {}).get('/V', '')
|
||||
if last_name and first_name:
|
||||
return f"{last_name}_{first_name}.pdf"
|
||||
else:
|
||||
return None
|
||||
except Exception as e:
|
||||
return None
|
||||
|
||||
@app.route('/', methods=['GET', 'POST'])
|
||||
def upload_files():
|
||||
if request.method == 'POST':
|
||||
files = request.files.getlist('files')
|
||||
processed_files = []
|
||||
temp_extract_dirs = []
|
||||
for file in files:
|
||||
if file and allowed_file(file.filename):
|
||||
filename = secure_filename(file.filename)
|
||||
filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
|
||||
file.save(filepath)
|
||||
if filename.lower().endswith('.zip'):
|
||||
# Extract ZIP
|
||||
extract_dir = os.path.join(app.config['UPLOAD_FOLDER'], 'extract_' + os.path.splitext(filename)[0])
|
||||
os.makedirs(extract_dir, exist_ok=True)
|
||||
temp_extract_dirs.append(extract_dir)
|
||||
with zipfile.ZipFile(filepath, 'r') as zip_ref:
|
||||
zip_ref.extractall(extract_dir)
|
||||
# Process PDFs in extract_dir
|
||||
for root, dirs, files_in_zip in os.walk(extract_dir):
|
||||
for f in files_in_zip:
|
||||
if f.lower().endswith('.pdf'):
|
||||
pdf_path = os.path.join(root, f)
|
||||
new_name = extract_names_from_pdf(pdf_path)
|
||||
if new_name:
|
||||
new_path = os.path.join(root, new_name)
|
||||
counter = 1
|
||||
while os.path.exists(new_path):
|
||||
base, ext = os.path.splitext(new_name)
|
||||
new_path = os.path.join(root, f"{base}_{counter}{ext}")
|
||||
counter += 1
|
||||
os.rename(pdf_path, new_path)
|
||||
processed_files.append(('success', os.path.relpath(new_path, extract_dir), new_path))
|
||||
else:
|
||||
processed_files.append(('skipped', f, None))
|
||||
else:
|
||||
# Single PDF
|
||||
new_name = extract_names_from_pdf(filepath)
|
||||
if new_name:
|
||||
new_filepath = os.path.join(app.config['UPLOAD_FOLDER'], new_name)
|
||||
counter = 1
|
||||
while os.path.exists(new_filepath):
|
||||
base, ext = os.path.splitext(new_name)
|
||||
new_filepath = os.path.join(app.config['UPLOAD_FOLDER'], f"{base}_{counter}{ext}")
|
||||
counter += 1
|
||||
os.rename(filepath, new_filepath)
|
||||
processed_files.append(('success', os.path.basename(new_filepath), new_filepath))
|
||||
else:
|
||||
processed_files.append(('skipped', filename, None))
|
||||
|
||||
# Create ZIP of all successful renames
|
||||
successful_files = [p for p in processed_files if p[0] == 'success']
|
||||
if successful_files:
|
||||
zip_output = os.path.join(app.config['UPLOAD_FOLDER'], 'renamed_files.zip')
|
||||
with zipfile.ZipFile(zip_output, 'w') as zip_out:
|
||||
for _, rel_name, full_path in successful_files:
|
||||
zip_out.write(full_path, rel_name)
|
||||
zip_name = 'renamed_files.zip'
|
||||
else:
|
||||
zip_name = None
|
||||
|
||||
skipped = [p[1] for p in processed_files if p[0] == 'skipped']
|
||||
|
||||
return render_template_string('''
|
||||
<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<title>Results</title>
|
||||
<style>
|
||||
body { font-family: Arial, sans-serif; background-color: #f4f4f4; margin: 0; padding: 20px; text-align: center; }
|
||||
img { margin-bottom: 20px; }
|
||||
h1 { color: #333; }
|
||||
p { font-size: 18px; }
|
||||
a { color: #007bff; text-decoration: none; padding: 10px 20px; background-color: #007bff; color: white; border-radius: 5px; }
|
||||
a:hover { background-color: #0056b3; }
|
||||
ul { list-style-type: none; padding: 0; }
|
||||
li { background-color: #fff; margin: 5px 0; padding: 10px; border-radius: 5px; box-shadow: 0 0 5px rgba(0,0,0,0.1); }
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<img src="/static/trbc-logo-full.png" alt="Logo" width="200">
|
||||
<h1>Processing Results</h1>
|
||||
{% if zip_name %}
|
||||
<p><a href="/download/{{ zip_name }}">Download Renamed PDFs (ZIP)</a></p>
|
||||
{% endif %}
|
||||
<h2>Skipped Files:</h2>
|
||||
<ul>
|
||||
{% for file in skipped %}
|
||||
<li>{{ file }}</li>
|
||||
{% endfor %}
|
||||
</ul>
|
||||
<a href="/">Upload more files</a>
|
||||
</body>
|
||||
</html>
|
||||
''', zip_name=zip_name, skipped=skipped)
|
||||
return render_template_string('''
|
||||
<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<title>PDF Renamer</title>
|
||||
<style>
|
||||
body { font-family: Arial, sans-serif; background-color: #f4f4f4; margin: 0; padding: 20px; text-align: center; }
|
||||
img { margin-bottom: 20px; }
|
||||
h1 { color: #333; }
|
||||
form { background-color: #fff; padding: 20px; border-radius: 10px; box-shadow: 0 0 10px rgba(0,0,0,0.1); display: inline-block; }
|
||||
input[type="file"] { margin: 10px 0; }
|
||||
input[type="submit"] { background-color: #28a745; color: white; border: none; padding: 10px 20px; border-radius: 5px; cursor: pointer; }
|
||||
input[type="submit"]:hover { background-color: #218838; }
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<img src="/static/trbc-logo-full.png" alt="Logo" width="200">
|
||||
<h1>Upload PDF Files or ZIP to Rename</h1>
|
||||
<form method="post" enctype="multipart/form-data">
|
||||
<input type="file" name="files" multiple accept=".pdf,.zip">
|
||||
<br><br>
|
||||
<input type="submit" value="Process PDFs">
|
||||
</form>
|
||||
</body>
|
||||
</html>
|
||||
''')
|
||||
|
||||
@app.route('/download/<filename>')
|
||||
def download_file(filename):
|
||||
return send_from_directory(app.config['UPLOAD_FOLDER'], filename, as_attachment=True)
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(debug=True, port=3000)
|
||||
Reference in new issue
Block a user