# coding=utf-8
import json
import _mysql
import time
from framework import *
from database import *
from post import *
def api(self, path_split):
if len(path_split) > 2:
try:
self.output = api_process(self, path_split)
except APIError, e:
self.output = api_error("error", e.message)
except UserError, e:
self.output = api_error("failed", e.message)
except Exception, e:
import sys, traceback
exc_type, exc_value, exc_traceback = sys.exc_info()
detail = ["%s : %s : %s : %s" % (os.path.basename(o[0]),o[1],o[2],o[3]) for o in traceback.extract_tb(exc_traceback)]
self.output = api_error("exception", str(e), str(type(e)), detail)
else:
self.output = api_error("error", "No method specified")
def api_process(self, path_split):
formdata = self.formdata
ip = self.environ["REMOTE_ADDR"]
t = time.time()
method = path_split[2]
#bans = ['181.72.116.62']
bans = []
if ip in bans:
raise APIError, "You have been blacklisted."
#with open('../api_log.txt', 'a') as f:
# logstr = "[%s] %s: %s\n" % (formatTimestamp(t), ip, repr(path_split))
# f.write(logstr)
values = {'state': 'success'}
if method == 'boards':
boards = FetchAll('SELECT dir, name, board_type, allow_images, allow_image_replies, maxsize FROM `boards` WHERE `secret`=0 ORDER BY `name` ASC')
values['boards'] = boards
for board in values['boards']:
board['board_type'] = int(board['board_type'])
board['allow_images'] = int(board['allow_images'])
board['allow_image_replies'] = int(board['allow_image_replies'])
board['maxsize'] = int(board['maxsize'])
elif method == 'last':
data_limit = formdata.get('limit')
data_since = formdata.get('since')
limit = 10
since = 0
if data_limit:
try:
limit = int(data_limit)
except ValueError:
raise APIError, "Limit must be numeric"
if data_since:
try:
since = int(data_since)
except ValueError:
raise APIError, "Since must be numeric"
if limit > 50:
raise APIError, "Maximum limit is 50"
sql = "SELECT posts.id, boards.dir, timestamp, timestamp_formatted, posts.name, tripcode, email, posts.subject, posts.message, file, file_size, image_height, image_width, thumb, thumb_width, thumb_height, parentid FROM posts INNER JOIN boards ON boardid = boards.id WHERE timestamp > %d AND IS_DELETED = 0 AND boards.secret = 0 ORDER BY timestamp DESC LIMIT %d" % (since, limit)
values['posts'] = FetchAll(sql)
for post in values['posts']:
post['id'] = int(post['id'])
post['timestamp'] = int(post['timestamp'])
post['parentid'] = int(post['parentid'])
post['file_size'] = int(post['file_size'])
post['image_width'] = int(post['image_width'])
post['image_height'] = int(post['image_height'])
post['thumb_width'] = int(post['thumb_width'])
post['thumb_height'] = int(post['thumb_height'])
post['message'] = post['message'].decode('utf-8', 'replace')
elif method == 'lastage':
data_limit = formdata.get('limit')
data_time = formdata.get('time', 0)
limit = 30
if data_limit:
try:
limit = int(data_limit)
except ValueError:
raise APIError, "Limit must be numeric"
if limit > 50:
raise APIError, "Maximum limit is 50"
threads = getLastAge(limit)
if threads[0]['bumped'] > int(data_time):
values['threads'] = threads
else:
values['threads'] = []
elif method == 'list':
data_board = formdata.get('dir')
data_offset = formdata.get('offset')
data_limit = formdata.get('limit')
data_replies = formdata.get('replies')
offset = 0
limit = 10
numreplies = 2
if not data_board:
raise APIError, "Missing parameters"
if data_limit:
try:
limit = int(data_limit)
except ValueError:
raise APIError, "Limit must be numeric"
if data_offset:
try:
offset = int(data_offset)
except ValueError:
raise APIError, "Offset must be numeric"
if data_replies:
try:
numreplies = int(data_replies)
except ValueError:
raise APIError, "Replies must be numeric"
if data_replies and limit > 30:
raise APIError, "Maximum limit is 30"
board = setBoard(data_board)
#sql = "SELECT id, timestamp, bumped, timestamp_formatted, name, tripcode, email, subject, message, file, thumb FROM posts WHERE boardid = %s AND parentid = 0 AND IS_DELETED = 0 ORDER BY bumped DESC LIMIT %d" % (board['id'], limit)
sql = "SELECT p.id, p.timestamp, p.bumped, p.expires, p.expires_formatted, p.timestamp_formatted, p.name, p.tripcode, p.email, p.subject, p.message, p.file, p.file_size, p.image_width, p.image_height, p.thumb, p.thumb_height, p.thumb_width, p.locked, coalesce(x.count,0) AS total_replies, coalesce(x.files,0) AS total_files FROM `posts` AS p LEFT JOIN (SELECT parentid, count(1) as count, count(nullif(file, '')) as files FROM `posts` WHERE boardid = %(board)s GROUP BY parentid) AS x ON p.id=x.parentid WHERE p.parentid = 0 AND p.boardid = %(board)s AND p.IS_DELETED = 0 ORDER BY `bumped` DESC LIMIT %(limit)d OFFSET %(offset)d" % {'board': board["id"], 'limit': limit, 'offset': offset}
threads = FetchAll(sql)
if numreplies:
for thread in threads:
lastreplies = FetchAll("SELECT id, timestamp, timestamp_formatted, name, tripcode, email, subject, message, file, file_size, image_height, image_width, thumb, thumb_width, thumb_height, IS_DELETED FROM `posts` WHERE parentid = %s AND boardid = %s ORDER BY `timestamp` DESC LIMIT %d" % (thread['id'], board['id'], numreplies))
lastreplies = lastreplies[::-1]
thread['id'] = int(thread['id'])
thread['timestamp'] = int(thread['timestamp'])
thread['bumped'] = int(thread['bumped'])
thread['expires'] = int(thread['expires'])
thread['total_replies'] = int(thread['total_replies'])
thread['total_files'] = int(thread['total_files'])
thread['file_size'] = int(thread['file_size'])
thread['image_width'] = int(thread['image_width'])
thread['image_height'] = int(thread['image_height'])
thread['thumb_width'] = int(thread['thumb_width'])
thread['thumb_height'] = int(thread['thumb_height'])
thread['locked'] = int(thread['locked'])
thread['replies'] = []
for post in lastreplies:
post['IS_DELETED'] = int(post['IS_DELETED'])
post['id'] = int(post['id'])
post['timestamp'] = int(post['timestamp'])
if post['IS_DELETED']:
empty_post = {'id': post['id'],
'IS_DELETED': post['IS_DELETED'],
'timestamp': post['timestamp'],
}
thread['replies'].append(empty_post)
else:
post['file_size'] = int(post['file_size'])
post['image_width'] = int(post['image_width'])
post['image_height'] = int(post['image_height'])
post['thumb_width'] = int(post['thumb_width'])
post['thumb_height'] = int(post['thumb_height'])
post['message'] = post['message'].decode('utf-8', 'replace')
thread['replies'].append(post)
values['threads'] = threads
elif method == 'thread':
data_board = formdata.get('dir')
data_threadid = formdata.get('id')
data_threadts = formdata.get('ts')
data_offset = formdata.get('offset')
data_limit = formdata.get('limit')
data_striphtml = formdata.get('nohtml')
striphtml = False
offset = 0
limit = 1000
if not data_board or (not data_threadid and not data_threadts):
raise APIError, "Missing parameters"
if data_limit:
try:
limit = int(data_limit)
except ValueError:
raise APIError, "Limit must be numeric"
if data_offset:
try:
offset = int(data_offset)
except ValueError:
raise APIError, "Offset must be numeric"
if data_striphtml:
if int(data_striphtml) == 1:
striphtml = True
board = setBoard(data_board)
search_field = 'id'
search_val = 0
try:
search_val = int(data_threadid)
except (ValueError, TypeError):
pass
try:
search_val = int(data_threadts)
search_field = 'timestamp'
except (ValueError, TypeError):
pass
if not search_val:
raise APIError, "No thread ID"
op_post = FetchOne("SELECT id, timestamp, subject, locked FROM posts WHERE `%s` = '%d' AND boardid = '%s' AND parentid = 0" % (search_field, search_val, board["id"]))
if not op_post:
raise APIError, "Not a thread"
values['id'] = int(op_post['id'])
values['timestamp'] = int(op_post['timestamp'])
values['subject'] = op_post['subject']
values['locked'] = int(op_post['locked'])
total_replies = int(FetchOne("SELECT COUNT(1) FROM posts WHERE boardid = '%s' AND parentid = '%d'" % (board["id"], values['id']), 0)[0])
values['total_replies'] = total_replies
sql = "SELECT id, parentid, timestamp, timestamp_formatted, name, tripcode, email, subject, message, file, file_size, image_width, image_height, thumb, thumb_width, thumb_height, IS_DELETED FROM posts WHERE boardid = %s AND (parentid = %s OR id = %s) ORDER BY id ASC LIMIT %d OFFSET %d" % (_mysql.escape_string(board['id']), values['id'], values['id'], limit, offset)
posts = FetchAll(sql)
values['posts'] = []
for post in posts:
post['IS_DELETED'] = int(post['IS_DELETED'])
post['id'] = int(post['id'])
post['parentid'] = int(post['parentid'])
post['timestamp'] = int(post['timestamp'])
if post['IS_DELETED']:
empty_post = {'id': post['id'],
'IS_DELETED': post['IS_DELETED'],
'parentid': post['parentid'],
'timestamp': post['timestamp'],
}
values['posts'].append(empty_post)
else:
post['file_size'] = int(post['file_size'])
post['image_width'] = int(post['image_width'])
post['image_height'] = int(post['image_height'])
post['thumb_width'] = int(post['thumb_width'])
post['thumb_height'] = int(post['thumb_height'])
post['message'] = post['message'].decode('utf-8', 'replace')
if striphtml:
post['message'] = post['message'].replace("<br />", " ")
post['message'] = re.compile(r"<[^>]*?>", re.DOTALL | re.IGNORECASE).sub("", post['message'])
values['posts'].append(post)
elif method == 'get':
data_board = formdata.get('dir')
data_parentid = formdata.get('thread')
data_postid = formdata.get('id')
data_postnum = formdata.get('num')
if not data_board and (not data_postid or (not data_postnum and not data_parentid)):
raise APIError, "Missing parameters"
board = setBoard(data_board)
postid = 0
if data_postnum:
data_postid = getID(data_parentid, data_postid)
try:
postid = int(data_postid)
except ValueError:
raise APIError, "Post ID must be numeric"
post = FetchOne("SELECT id, parentid, timestamp, timestamp_formatted, name, tripcode, email, subject, message, file, file_size, image_width, image_height, thumb, thumb_width, thumb_height, IS_DELETED FROM posts WHERE `id`='%d' AND boardid='%s'" % (postid, board["id"]))
if not post:
raise APIError, "Post ID cannot be found"
values['posts'] = []
post['IS_DELETED'] = int(post['IS_DELETED'])
post['id'] = int(post['id'])
post['parentid'] = int(post['parentid'])
post['timestamp'] = int(post['timestamp'])
if post['IS_DELETED']:
empty_post = {'id': post['id'],
'IS_DELETED': post['IS_DELETED'],
'parentid': post['parentid'],
'timestamp': post['timestamp'],
}
values['posts'].append(empty_post)
else:
post['file_size'] = int(post['file_size'])
post['image_width'] = int(post['image_width'])
post['image_height'] = int(post['image_height'])
post['thumb_width'] = int(post['thumb_width'])
post['thumb_height'] = int(post['thumb_height'])
post['message'] = post['message'].decode('utf-8', 'replace')
values['posts'].append(post)
elif method == 'delete':
data_board = formdata.get('dir')
data_postid = formdata.get('id')
data_imageonly = formdata.get('imageonly')
data_password = formdata.get('password')
if not data_board or not data_postid or not data_password:
raise APIError, "Missing parameters"
imageonly = False
board = setBoard(data_board)
try:
postid = int(data_postid)
except ValueError:
raise APIError, "Post ID must be numeric"
if data_imageonly and data_imageonly == 1:
imageonly = True
deletePost(postid, data_password, board['recyclebin'], imageonly)
elif method == 'post':
boarddir = formdata.get('board')
if not boarddir:
raise APIError, "Missing parameters"
parent = formdata.get('parent')
trap1 = formdata.get('name', '')
trap2 = formdata.get('email', '')
name = formdata.get('fielda', '')
email = formdata.get('fieldb', '')
subject = formdata.get('subject', '')
message = formdata.get('message', '')
file = formdata.get('file')
file_original = formdata.get('file_original')
spoil = formdata.get('spoil')
oek_file = formdata.get('oek_file')
password = formdata.get('password', '')
noimage = formdata.get('noimage')
mobile = ("mobile" in formdata.keys())
# call post function
(post_url, ttaken) = self.make_post(ip, boarddir, parent, trap1, trap2, name, email, subject, message, file, file_original, spoil, oek_file, password, noimage, mobile)
values['post_url'] = post_url
values['time_taken'] = ttaken
else:
raise APIError, "Invalid method"
values['time'] = int(t)
#values['time_taken'] = time.time() - t
return json.dumps(values, sort_keys=True, separators=(',',':'))
def api_error(errtype, msg, type=None, detail=None):
values = {'state': errtype, 'message': msg}
if type:
values['type'] = type
if detail:
values['detail'] = detail
return json.dumps(values)
class APIError(Exception):
pass
