main.py

import argparse
import os

import flask
from flask import Flask, render_template, jsonify, send_from_directory, url_for, request
from werkzeug.utils import secure_filename, redirect

from website import system
from website import articles

from flask.ext.uploads import *

app = Flask(__name__)


@app.route("/index")
@app.route("/")
def index():
	return render_template("index.html")


@app.route("/contact", methods=["GET", "POST"])
def contact():
	return render_template("contact.html")


@app.route("/_ram")
def get_ram():
	response = {
		"total": round(system.get_total_ram()),
		"server": round(system.get_process_ram()),
		"other": round(system.get_other_ram()),
	}
	return jsonify(response)


@app.route("/_cpu")
def get_cpu():
	response = {
		"percentage": [round(system.get_cpu())]
	}
	return jsonify(response)


# pages
@app.route("/stats")
def stats_page():
	return render_template("stats.html")


@app.route("/projects")
def projects_page():
	projects = [
		{"title": "Useful Links", "url": "/articles/usefulLinks", "img": "static/img/tux.png", "text": "Collection of useful articles"},
		{"title": "Website Development", "url": "/articles/websiteDevelopment1", "img": "static/img/html-tag.png", "text": "The history of this website"}
	]
	return render_template("projects.html", projects=projects)


@app.route("/articles")
def articles_page():
	recentArticles = reversed(articles.get_published())
	return render_template("articles.html", articles=recentArticles)


@app.route("/articles/<name>")
def getArticle(name):
	info = articles.get_article_data_by_url(name)
	html = articles.get_article_html(name)

	related = []
	try:
		for article in info["related"]:
			try:
				related.append(articles.get_article_data_by_url(article))
			except FileNotFoundError:
				pass
	except KeyError:
		pass

	if len(related) == 0:
		related = None

	print("Displaying {}".format(info))
	return render_template("article.html", info=info, html=html, related=related)


@app.route("/search")
def getSearchResults():
	return jsonify


@app.route("/woggles")
def woggles():
	return render_template("woggles.html")


@app.route("/files/<path:path>")
def serve_file(path):
	dir = os.path.join(os.getcwd(), "files")
	print(dir)
	return send_from_directory(dir, path)


@app.route("/upload", methods=["POST"])
def upload():
	if request.method == "POST":
		file = request.files["file"]
		filename = secure_filename(file.filename)
		path = os.path.join("files", filename)
		if not os.path.isfile(path):
			print(path, "doesn't exist")
			file.save(path)
			return redirect("/woggles")
		else:
			return "File with that name already exists", 418


if __name__ == "__main__":
	parser = argparse.ArgumentParser()
	parser.add_argument("-d", dest="debug", type=bool)
	args = parser.parse_args()

	if args.debug:
		app.debug = True

	app.run()