# coding=utf-8
import os
import cgi
import random
from database import *
from settings import Settings
from framework import *
from formatting import *
from template import *
from post import *
import netaddr
def oekaki(self, path_split):
"""
Este script hace todo lo que tiene que hacer con los
archivos de Oekaki.
"""
page = ''
skiptemplate = False
if len(path_split) > 2:
# Inicia el applet. Lo envia luego a este mismo script, a "Finish".
if path_split[2] == 'paint':
# Veamos que applet usar
applet = self.formdata['oek_applet'].split('|')
applet_name = applet[0]
if len(applet) > 1 and applet[1] == 'y':
applet_str = 'pro'
else:
applet_str = ''
if len(applet) > 2 and applet[2] == 'y':
use_selfy = True
else:
use_selfy = False
# Obtenemos el board
board = setBoard(self.formdata['board'])
if not board['allow_oekaki']:
raise UserError('Esta sección no soporta oekaki.')
# Veamos a quien le estamos respondiendo
try:
parentid = int(self.formdata['parent'])
except:
parentid = 0
# Vemos si el usuario quiere una animacion
if 'oek_animation' in self.formdata:
animation = True
animation_str = 'animation'
else:
animation = False
animation_str = ''
# Nos aseguramos que la entrada es numerica
try:
width = int(self.formdata['oek_x'])
height = int(self.formdata['oek_y'])
except:
raise UserError('Valores de tamaño inválidos (%s)' % repr(
self.formdata))
params = {
'dir_resource': Settings.BOARDS_URL + 'oek_temp/',
'tt.zip': 'tt_def.zip',
'res.zip': 'res.zip',
'MAYSCRIPT': 'true',
'scriptable': 'true',
'tools': applet_str,
'layer_count': '5',
'undo': '90',
'undo_in_mg': '15',
'url_save': Settings.BOARDS_URL + 'oek_temp/save.py?applet=shi'+applet_str,
'poo': 'false',
'send_advance': 'true',
'send_language': 'utf8',
'send_header': '',
'send_header_image_type': 'false',
'thumbnail_type': animation_str,
'image_jpeg': 'false',
'image_size': '92',
'compress_level': '4'
}
if 'oek_edit' in self.formdata:
# Si hay que editar, cargar la imagen correspondiente en el canvas
pid = int(self.formdata['oek_edit'])
post = FetchOne(
'SELECT id, file, image_width, image_height FROM posts WHERE id = %d AND boardid = %s' % (pid, board['id']))
editfile = Settings.BOARDS_URL + \
board['dir'] + '/src/' + post['file']
params['image_canvas'] = edit
params['image_width'] = file['image_width']
params['image_height'] = file['image_height']
width = int(file['image_width'])
height = int(file['image_height'])
else:
editfile = None
params['image_width'] = str(width)
params['image_height'] = str(height)
if 'canvas' in self.formdata:
editfile = self.formdata['canvas']
# Darle las dimensiones al exit script
params['url_exit'] = Settings.CGI_URL + \
'oekaki/finish/' + board['dir'] + '/' + str(parentid)
page += renderTemplate("paint.html", {'applet': applet_name, 'edit': editfile, 'replythread': parentid,
'width': width, 'height': height, 'params': params, 'selfy': use_selfy})
elif path_split[2] == 'save':
# path splits:
# 3: Board
# 4: Data format
board = setBoard(path_split[3])
ip = int(netaddr.IPAddress(self.ip))
fname = os.path.join(Settings.IMAGES_DIR,
board['dir'], "temp", str(ip) + ".png")
if path_split[4] == 'b64':
page = write_from_base64(fname, self.formdata['image'])
elif path_split[4] == 'paintbbs':
page = write_from_shi(fname, self.environ["wsgi.input"])
elif path_split[2] == 'finish':
# path splits:
# 3: Board
# 4: Parentid
if len(path_split) >= 5:
# Al terminar de dibujar, llegamos aqui. Damos la opcion de postearlo.
board = setBoard(path_split[3])
try:
parentid = int(path_split[4])
except:
parentid = None
if not board['allow_oekaki']:
raise UserError('Esta sección no soporta oekaki.')
ts = int(time.time())
ip = int(netaddr.IPAddress(self.ip))
fname = os.path.join(Settings.IMAGES_DIR,
board['dir'], "temp", str(ip) + ".png")
oek = 'no'
if 'filebase' in self.formdata:
write_from_base64(fname, self.formdata['filebase'])
if os.path.isfile(fname):
oek = ip
try:
timetaken = timestamp() - int(path_split[5][:-2])
except:
timetaken = 0
page += renderTemplate("board.html", {
"threads": None, "oek_finish": oek, "replythread": parentid, "ts": ts})
elif path_split[2] == 'animation':
try:
board = setBoard(path_split[3])
file = int(path_split[4])
except:
raise UserError('Board o archivo de animación inválido.')
params = {
'pch_file': Settings.BOARDS_URL + board['dir'] + '/src/' + str(file) + '.pch',
'run': 'true',
'buffer_progress': 'false',
'buffer_canvas': 'true',
'speed': '2',
'res.zip': Settings.BOARDS_URL + 'oek_temp/res/' + 'res.zip',
'tt.zip': Settings.BOARDS_URL + 'oek_temp/res/' + 'tt.zip',
'tt_size': '31'
}
page += '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">' + \
'<html xmlns="http://www.w3.org/1999/xhtml">\n<head><style type="text/css">html, body{margin: 0; padding: 0;height:100%;} .full{width:100%;height:100%;}</style>\n<title>Bienvenido a Internet | Oekaki</title>\n</head>\n' + \
'<body bgcolor="#CFCFFF" text="#800000" link="#003399" vlink="#808080" alink="#11FF11">\n' + \
'<table cellpadding="0" cellspacing="0" class="full"><tr><td class="full">\n'
page += '<applet name="pch" code="pch2.PCHViewer.class" archive="' + \
Settings.BOARDS_URL + 'oek_temp/PCHViewer123.jar" width="100%" height="100%">'
for key in params:
page += '<param name="' + key + '" value="' + \
cleanString(params[key]) + '" />' + "\n"
page += '<div align="center">Java must be installed and enabled to use this applet. Please refer to our Java setup tutorial for more information.</div>'
page += '</applet>\n</td></tr></table>\n</body>\n</html>'
if not skiptemplate:
self.output = page
def write_from_base64(fname, data):
# Skip header
if data.startswith("data:image/png;base64,"):
data = data[22:]
data = data.replace(' ', '+')
data = decodeb64(data)
with open(fname, 'wb') as f:
f.write(data)
return "OK"
def write_from_shi(fname, fp):
# Check data type
type = fp.read(1)
if type != b'P':
return "UNSUPPORTED"
# Read header
headerLength = int(fp.read(8))
header = fp.read(headerLength)
# Read image data
imgLength = int(fp.read(8))
fp.read(2) # TODO: seek() isn't working for some reason. Debug.
img = fp.read(imgLength)
# Write image
with open(fname, 'wb') as f:
f.write(img)
return "OK"
