#!/usr/bin/env python3
from htmlgen import *
import traceback
import storage
import cgi
import random
import base64
from io import BytesIO
from captcha.image import ImageCaptcha
import time, string
print("Content-type: text/html\r\n\r\n")
form=cgi.FieldStorage()
aid=int(form["aid"].value)
try:
seed=int(form["seed"].value)
random.seed(seed)
checkstr=list("".join(random.choice(string.digits+string.ascii_lowercase) for _ in range(5)))
if not checkstr == form["checkstr"].value:
print("Captcha's wrong")
raise SyntaxError()
username=form["username"].value
try:
message=form["message"].value
except:
raise SyntaxError()
timestamp=int(time.time())
data={"name":"","markdown":message,"author":username,"date":timestamp}
storage.append("comments-%i"%aid,data)
except KeyError:
pass
seed=random.SystemRandom().randint(0,2**24)
random.seed(seed)
checkstr=list("".join(random.choice(string.digits+string.ascii_lowercase) for _ in range(5)))
image = ImageCaptcha()
capt = image.generate(checkstr)
html=htmlgen.HTMLgen(pagelayout.getLayoutXML().decode('utf-8'),"Comments")
html.appendHTML(""%seed)
html.appendHTML("
")
html.appendHTML("
")
html.appendHTML("![\"Captcha](\"data:image/png;base64,%s\")
" % base64.b64encode(capt.getvalue()).decode("UTF-8"))
html.appendHTML("")
html.appendHTML("")
count=storage.count("comments-%i"%aid)
for i in range(count):
html.addArticle(**(storage.get("comments-%i"%aid,i)))
print("")
print(html.renderSite())