101 lines
3.8 KiB
Python
101 lines
3.8 KiB
Python
from flask import Flask, render_template, send_from_directory
|
|
from flask_frozen import Freezer
|
|
import os
|
|
|
|
app = Flask(__name__)
|
|
freezer = Freezer(app)
|
|
|
|
app.config['FREEZER_RELATIVE_URLS'] = True
|
|
app.config['FREEZER_IGNORE_MIMETYPE_WARNINGS'] = True
|
|
|
|
@app.route('/')
|
|
@app.route('/index.html')
|
|
def index():
|
|
categories = os.listdir('projects/')
|
|
projects = []
|
|
for category in categories:
|
|
for project in os.listdir('projects/'+category):
|
|
# projects.append(project)
|
|
projects.append([category, project, os.listdir('projects/'+category+'/'+project)[0]])
|
|
return render_template('index.html', categories=categories, projects=projects)
|
|
|
|
@app.route('/projects')
|
|
@app.route('/projects/')
|
|
def show_projects():
|
|
categories=os.listdir('projects/')
|
|
return render_template('project_category.html', categories=categories)
|
|
|
|
@freezer.register_generator
|
|
def category_url_generator():
|
|
for category in os.listdir(os.getcwd() + '/projects/'):
|
|
if os.path.isdir(os.getcwd() + '/projects/' + category):
|
|
yield 'show_project_category', {'category':category}
|
|
|
|
@app.route('/projects/<category>.html')
|
|
def show_project_category(category):
|
|
categories=os.listdir('projects/')
|
|
projects = os.listdir('projects/' + category)
|
|
#print projects
|
|
if category in categories:
|
|
return render_template('project_category.html', category=category, projects=projects)
|
|
else:
|
|
return "Error, category not found!"
|
|
|
|
@freezer.register_generator
|
|
def project_url_generator():
|
|
for category in os.listdir(os.getcwd() + '/projects/'):
|
|
if os.path.isdir(os.getcwd() + '/projects/' + category):
|
|
for i in os.listdir(os.getcwd() + '/projects/' + category):
|
|
if os.path.isdir(os.getcwd() + '/projects/' + category + '/' + i):
|
|
#yield ('show_project', {'category': category, 'project_name': i})
|
|
yield ('show_project', {'category':category, 'project_name':i})
|
|
|
|
@app.route('/projects/<category>/<project_name>.html')
|
|
def show_project(category, project_name):
|
|
categories=os.listdir('projects')
|
|
projects = os.listdir('projects/' + category)
|
|
photo_dir = "projects/" + category + '/' + project_name
|
|
photos = os.listdir(photo_dir)
|
|
if project_name in projects:
|
|
return render_template('project.html', category=category, project_name=project_name, photos=photos, photo_dir=photo_dir)
|
|
else:
|
|
return "Error, project not found!"
|
|
|
|
@freezer.register_generator
|
|
def photo_url_generator():
|
|
projects_dir = os.getcwd() + '/projects/'
|
|
for category in os.listdir(projects_dir):
|
|
category_dir = projects_dir + category
|
|
if os.path.isdir(category_dir):
|
|
for i in os.listdir(category_dir):
|
|
project_dir = category_dir + '/' + i
|
|
if os.path.isdir(project_dir):
|
|
for photo in os.listdir(project_dir):
|
|
yield ('show_project_file', {'category':category, 'project_name':i, 'filename':photo})
|
|
|
|
|
|
@app.route('/projects/<category>/<project_name>/<filename>')
|
|
def show_project_file(category, project_name, filename):
|
|
return send_from_directory('projects/' + category + '/'+ project_name + '/', filename)
|
|
|
|
@app.route('/about.html')
|
|
def about():
|
|
return render_template('aboutus.html', title="About Us - JPs Contracting")
|
|
|
|
@app.route('/portfolio')
|
|
def portfolio():
|
|
categories = os.listdir('projects/')
|
|
projects = []
|
|
for category in categories:
|
|
for project in os.listdir('projects/'+category):
|
|
# projects.append(project)
|
|
projects.append([category, project, os.listdir('projects/'+category+'/'+project)[0]])
|
|
return render_template('index.html', title="JPs Contracting - Portfolio", categories=categories, projects=projects)
|
|
|
|
@app.route('/contact')
|
|
def contact():
|
|
return render_template('contactus.html', title="Contact Us - JPs Contracting")
|
|
|
|
if __name__ == '__main__':
|
|
freezer.freeze()
|