# coding=utf-8
import math
import os
import shutil
import time
import threading
import queue
import formatting
import logging
import html
from database import *
from template import *
from settings import Settings
from framework import *
class Post(object):
def __init__(self, boardid=0):
self.post = {
"boardid": boardid,
"parentid": 0,
"name": "",
"tripcode": "",
"email": "",
"subject": "",
"message": "",
"password": "",
"file": "",
"file_hex": "",
"file_size": 0,
"thumb": "",
"image_width": 0,
"image_height": 0,
"thumb_width": 0,
"thumb_height": 0,
"ip": "",
"IS_DELETED": 0,
"timestamp_formatted": "",
"timestamp": 0,
"bumped": 0,
"locked": 0,
"last": 0,
"expires": 0,
"expires_formatted": "",
"expires_alert": 0,
"length": 0,
}
def __getitem__(self, key):
return self.post[key]
def __setitem__(self, key, value):
self.post[key] = value
def __iter__(self):
return self.post
def insert(self):
logging.info("Insertando Post")
post_keys = []
post_templates = []
post_values = []
for key, value in self.post.items():
post_keys.append(key)
if key == 'ip':
template = "INET6_ATON(%s)"
else:
template = "%s"
post_templates.append(template)
post_values.append(value)
return InsertDb("INSERT INTO `posts` (`%s`) VALUES (%s)" % (
"`, `".join(post_keys),
", ".join(post_templates)
), post_values)
class RegenerateThread(threading.Thread):
def __init__(self, threadid, request_queue):
threading.Thread.__init__(self, name="RegenerateThread-%d" % (threadid,))
self.request_queue = request_queue
self.board = Settings._.BOARD
def run(self):
Settings._.BOARD = self.board
while 1:
action = self.request_queue.get()
if action is None:
break
if action == "front":
regenerateFrontPages()
else:
regenerateThreadPage(action)
def threadNumReplies(post):
"""
Get how many replies a thread has
"""
board = Settings._.BOARD
num = FetchOne("SELECT COUNT(1) AS 'count' FROM `posts` WHERE `parentid` = %s AND `boardid` = %s" % (post, board['id']))
return int(num["count"]) + 1
def get_parent_post(post_id, board_id):
post = FetchOne("SELECT `id`, `email`, `message`, `locked`, `subject`, `timestamp`, `bumped`, `last`, `length` FROM `posts` WHERE `id` = %s AND `parentid` = 0 AND `IS_DELETED` = 0 AND `boardid` = %s LIMIT 1",
(post_id, board_id))
if post:
return post
else:
raise UserError(_("The ID of the parent post is invalid."))
def getThread(postid=0, mobile=False, timestamp=0):
board = Settings._.BOARD
total_bytes = 0
database_lock.acquire()
try:
if timestamp:
op_post = FetchOne("SELECT * FROM `posts` WHERE `timestamp` = %s AND `boardid` = %s AND parentid = 0 LIMIT 1", (timestamp, board["id"]))
else:
op_post = FetchOne("SELECT * FROM `posts` WHERE `id` = %s AND `boardid` = %s AND parentid = 0 LIMIT 1", (postid, board["id"]))
if op_post:
op_post['num'] = 1
if mobile:
op_post['timestamp_formatted'] = re.compile(r"\(.{1,3}\)", re.DOTALL | re.IGNORECASE).sub(" ", op_post['timestamp_formatted'])
thread = {"id": op_post["id"], "posts": [op_post], "omitted": 0}
#thread = {"id": op_post["id"], "posts": [op_post], "omitted": 0, "omitted_img": 0}
total_bytes += len(op_post["message"])+80
replies = FetchAll("SELECT IS_DELETED, email, file, file_size, id, image_height, image_width, message, name, subject, thumb, thumb_height, thumb_width, timestamp_formatted, tripcode, parentid, locked, expires, expires_alert, expires_formatted, timestamp FROM `posts` WHERE `parentid` = %s AND `boardid` = %s ORDER BY `id` ASC", (op_post["id"], board["id"]))
thread["length"] = 1
if replies:
for reply in replies:
thread["length"] += 1
reply['num'] = thread["length"]
if mobile:
reply['message'] = formatting.fixMobileLinks(reply['message'])
reply['timestamp_formatted'] = re.compile(r"\(.{1,3}\)", re.DOTALL | re.IGNORECASE).sub(" ", reply['timestamp_formatted'])
thread["posts"].append(reply)
total_bytes += len(reply["message"])+57
# An imageboard needs subject
if board["board_type"] in [1, 5]:
thread["timestamp"] = op_post["timestamp"]
thread["subject"] = op_post["subject"]
thread["message"] = op_post["message"]
thread["locked"] = op_post["locked"]
thread["size"] = "%d KB" % int(total_bytes / 1000)
else:
return None
finally:
database_lock.release()
return thread
def getID(threadid, postnum):
board = Settings._.BOARD
database_lock.acquire()
try:
posts = FetchAll("SELECT id FROM `posts` WHERE `parentid`= %s AND `boardid`= %s ORDER BY `id` ASC", (thread["id"], board["id"]))
if posts:
post = posts[int(postnum)-1]
postid = post["id"]
else:
return None
finally:
database_lock.release()
return postid
def shortenMsg(message, elid='0', elboard='0'):
"""
Intenta acortar el mensaje si es necesario
Algoritmo traducido desde KusabaX
"""
board = Settings._.BOARD
limit = 100 * int(board['numline'])
message_exploded = message.split('
')
if len(message) > limit or len(message_exploded) > int(board['numline']):
message_shortened = ''
for i in range(int(board['numline'])):
if i >= len(message_exploded):
break
message_shortened += message_exploded[i] + '
'
if len(message_shortened) > limit:
message_shortened = message_shortened[:limit]
message_shortened = formatting.close_html(message_shortened)
return True, message_shortened
else:
return False, message
def threadUpdated(postid):
"""
Shortcut to update front pages and thread page by passing a thread ID. Uses
the simple threading module to do both regenerateFrontPages() and
regenerateThreadPage() asynchronously
"""
# Use queues only if multithreading is enabled
if Settings.USE_MULTITHREADING:
request_queue = queue.Queue()
threads = [RegenerateThread(i, request_queue) for i in range(2)]
for t in threads:
t.start()
request_queue.put("front")
request_queue.put(postid)
for i in range(2):
request_queue.put(None)
for t in threads:
t.join
else:
regenerateFrontPages()
regenerateThreadPage(postid)
def regenerateFrontPages():
"""
Regenerates index.html and #.html for each page after that according to the number
of live threads in the database
"""
board = Settings._.BOARD
threads = []
if board['board_type'] == 1:
threads_to_fetch = int(board['numthreads'])
threads_to_limit = threads_to_fetch + 50
else:
if board['dir'] == 'o':
threads_to_fetch = threads_to_limit = int(board['numthreads'])*21
else:
threads_to_fetch = threads_to_limit = int(board['numthreads'])*11
database_lock.acquire()
try:
# fetch necessary threads and calculate how many posts we need
allthreads_query = "SELECT id, timestamp, subject, locked, length FROM `posts` WHERE `boardid` = %s AND parentid = 0 AND IS_DELETED = 0 ORDER BY `bumped` DESC, `id` ASC LIMIT %s"
allthreads = FetchAll(allthreads_query, (board["id"], threads_to_limit))
posts_to_fetch = 0
for t in allthreads[:threads_to_fetch]:
posts_to_fetch += int(t["length"])
more_threads = allthreads[threads_to_fetch:50]
# get the needed posts for the front page and order them
posts_query = "SELECT * FROM `posts` WHERE `boardid` = %s ORDER BY `bumped` DESC, CASE parentid WHEN 0 THEN id ELSE parentid END ASC, `id` ASC LIMIT %s"
posts = FetchAll(posts_query, (board["id"], posts_to_fetch))
threads = []
if posts:
thread = None
post_num = 0
for post in posts:
if not post["parentid"]:
skipThread = False
if not post["IS_DELETED"]:
# OP; Make new thread
if thread is not None:
thread["length"] = post_num
threads.append(thread)
post_num = post["num"] = 1
thread = {"id": post["id"], "timestamp": post["timestamp"], "subject": post["subject"], "locked": post["locked"], "posts": [post]}
else:
skipThread = True
else:
if not skipThread:
post_num += 1
post["num"] = post_num
thread["posts"].append(post)
if post_num:
thread["length"] = post_num
threads.append(thread)
finally:
database_lock.release()
pages = []
is_omitted = False
if len(threads) > 0:
# Todo : Make this better
if board['board_type'] == 1:
page_count = 1 # Front page only
threads_per_page = int(board['numthreads'])
else:
if board['dir'] == 'o':
front_limit = int(board['numthreads'])*21
else:
front_limit = int(board['numthreads'])*11
if len(threads) >= front_limit:
is_omitted = True
page_count = int(math.ceil(float(len(threads)) / float(int(board['numthreads']))))
threads_per_page = int(board['numthreads'])
for i in range(page_count):
pages.append([])
start = i * threads_per_page
end = start + threads_per_page
for thread in threads[start:end]:
pages[i].append(thread)
else:
page_count = 0
is_omitted = False
pages.append({})
page_num = 0
for pagethreads in pages:
regeneratePage(page_num, page_count, pagethreads, is_omitted, more_threads)
page_num += 1
def regeneratePage(page_num, page_count, threads, is_omitted=False, more_threads=[]):
"""
Regenerates a single page and writes it to .html
"""
board = Settings._.BOARD
for thread in threads:
replylimit = int(board['numcont'])
# Create reply list
parent = thread["posts"].pop(0)
replies = thread["posts"]
thread["omitted"] = 0
#thread["omitted_img"] = 0
# Omit posts
while(len(replies) > replylimit):
post = replies.pop(0)
thread["omitted"] += 1
#if post["file"]:
# thread["omitted_img"] += 1
# Remake thread with necessary replies only
replies.insert(0, parent)
thread["posts"] = replies
# Shorten messages
for post in thread["posts"]:
post["shortened"], post["message"] = shortenMsg(post["message"])
# Build page according to page number
if page_num == 0:
file_name = "index"
else:
file_name = str(page_num)
if board['board_type'] == 1:
templatename = "txt_board.html"
else:
templatename = "board.html"
page_rendered = renderTemplate(templatename, {"threads": threads, "pagenav": pageNavigator(page_num, page_count, is_omitted), "more_threads": more_threads})
f = open(Settings.ROOT_DIR + board["dir"] + "/" + file_name + ".html", "w")
try:
f.write(page_rendered)
finally:
f.close()
def threadList(mode=0):
board = Settings._.BOARD
if mode == 1:
mobile = True
maxthreads = 20
cutFactor = 100
elif mode == 2:
mobile = True
maxthreads = 1000
cutFactor = 50
elif mode == 3:
mobile = True
maxthreads = 1000
cutFactor = 100
else:
mobile = False
maxthreads = 1000
cutFactor = 70
if board['board_type'] == 1:
filename = "txt_threadlist.html"
full_threads = FetchAll("SELECT id, timestamp, timestamp_formatted, subject, length, last FROM `posts` WHERE parentid = 0 AND boardid = %(board)s AND IS_DELETED = 0 ORDER BY `bumped` DESC LIMIT %(limit)s" \
% {'board': board["id"], 'limit': maxthreads})
else:
filename = "threadlist.html"
full_threads = FetchAll("SELECT p.*, coalesce(x.count,1) AS length, coalesce(x.t,p.timestamp) AS last FROM `posts` AS p LEFT JOIN (SELECT parentid, count(1)+1 as count, max(timestamp) as t 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)s" \
% {'board': board["id"], 'limit': maxthreads})
# Generate threadlist
timestamps = []
for thread in full_threads:
if board['board_type'] == 1:
thread["timestamp_formatted"] = thread["timestamp_formatted"].split(" ")[0]
timestamps.append([thread["last"], formatTimestamp(thread["last"])])
if mobile:
timestamps[-1][1] = re.compile(r"\(.{1,3}\)", re.DOTALL | re.IGNORECASE).sub(" ", timestamps[-1][1])
else:
if len(thread['message']) > cutFactor:
thread['shortened'] = True
else:
thread['shortened'] = False
thread['message'] = thread['message'].replace('
', ' ')
thread['message'] = thread['message'].split("
")[0]
thread['message'] = re.compile(r"<[^>]*?>", re.DOTALL | re.IGNORECASE).sub('', thread['message'])
thread['message'] = thread['message'][:cutFactor]
thread['message'] = re.compile(r"&(.(?!;))*$", re.DOTALL | re.IGNORECASE).sub('', thread['message']) # Removes incomplete HTML entities
thread['timestamp_formatted'] = re.compile(r"\(.{1,3}\)", re.DOTALL | re.IGNORECASE).sub(" ", thread['timestamp_formatted'])
# Get last reply if in mobile mode
if mode == 1:
thread['timestamp_formatted'] = re.compile(r"\(.{1,3}\)", re.DOTALL | re.IGNORECASE).sub(" ", thread['timestamp_formatted'])
lastreply = FetchOne("SELECT * FROM `posts` WHERE parentid = %s AND boardid = %s AND IS_DELETED = 0 ORDER BY `timestamp` DESC LIMIT 1" % (thread['id'], board['id']))
if lastreply:
if len(lastreply['message']) > 60:
lastreply['shortened'] = True
else:
lastreply['shortened'] = False
lastreply['message'] = lastreply['message'].replace('
', ' ')
lastreply['message'] = lastreply['message'].split("
")[0]
lastreply['message'] = re.compile(r"<[^>]*?>", re.DOTALL | re.IGNORECASE).sub('', lastreply['message'])
lastreply['message'] = lastreply['message'][:60]
lastreply['message'] = re.compile(r"&(.(?!;))*$", re.DOTALL | re.IGNORECASE).sub('', lastreply['message']) # Removes incomplete HTML entities
lastreply['timestamp_formatted'] = re.compile(r"\(.{1,3}\)", re.DOTALL | re.IGNORECASE).sub(" ", lastreply['timestamp_formatted'])
thread["lastreply"] = lastreply
else:
thread["lastreply"] = None
elif mode == 2:
lastreply = FetchOne("SELECT timestamp_formatted FROM `posts` WHERE parentid = %s AND boardid = %s AND IS_DELETED = 0 ORDER BY `timestamp` DESC LIMIT 1" % (thread['id'], board['id']))
if lastreply:
lastreply['timestamp_formatted'] = re.compile(r"\(.{1,3}\)", re.DOTALL | re.IGNORECASE).sub(" ", lastreply['timestamp_formatted'])
thread["lastreply"] = lastreply
return renderTemplate(filename, {"more_threads": full_threads, "timestamps": timestamps, "mode": mode}, mobile)
def catalog(sort=''):
board = Settings._.BOARD
if board['board_type'] != 0:
raise UserError("No hay catálogo disponible para esta sección.")
cutFactor = 500
q_sort = '`bumped` DESC, `id` ASC'
if sort:
if sort == '1':
q_sort = '`timestamp` DESC'
elif sort == '2':
q_sort = '`timestamp` ASC'
elif sort == '3':
q_sort = '`length` DESC'
elif sort == '4':
q_sort = '`length` ASC'
threads = FetchAll("SELECT id, subject, message, length, thumb, expires_formatted FROM `posts` " +\
"WHERE parentid = 0 AND boardid = %(board)s AND IS_DELETED = 0 ORDER BY %(sort)s" \
% {'board': board["id"], 'sort': q_sort})
for thread in threads:
if len(thread['message']) > cutFactor:
thread['shortened'] = True
else:
thread['shortened'] = False
thread['message'] = thread['message'].replace('
', ' ')
thread['message'] = thread['message'].split("
")[0]
thread['message'] = re.compile(r"<[^>]*?>", re.DOTALL | re.IGNORECASE).sub('', thread['message'])
thread['message'] = thread['message'][:cutFactor]
thread['message'] = re.compile(r"&(.(?!;))*$", re.DOTALL | re.IGNORECASE).sub('', thread['message']) # Removes incomplete HTML entities
return renderTemplate("catalog.html", {"threads": threads, "i_sort": sort})
def regenerateThreadPage(postid):
"""
Regenerates /res/#.html for supplied thread id
"""
board = Settings._.BOARD
thread = getThread(postid)
if board['board_type'] in [1, 5]:
template_filename = "txt_thread.html"
outname = Settings.ROOT_DIR + board["dir"] + "/res/" + str(thread["timestamp"]) + ".html"
title_matome = thread['subject']
post_preview = cut_msg(thread['posts'][0]['message'], Settings.HOME_LASTPOSTS_LENGTH)
else:
template_filename = "board.html"
outname = Settings.ROOT_DIR + board["dir"] + "/res/" + str(postid) + ".html"
post_preview = cut_msg(thread['posts'][0]['message'], (Settings.HOME_LASTPOSTS_LENGTH - len(board['name'])) )
if thread['posts'][0]['subject'] != board['subject']:
title_matome = thread['posts'][0]['subject']
else:
title_matome = post_preview
page = renderTemplate(template_filename, {"threads": [thread], "replythread": postid, "matome": title_matome, "preview": post_preview}, False)
f = open(outname, "w")
try:
f.write(page)
finally:
f.close()
def threadPage(postid, mobile=False, timestamp=0):
board = Settings._.BOARD
if board['board_type'] in [1, 5]:
template_filename = "txt_thread.html"
else:
template_filename = "board.html"
thread = getThread(postid, mobile, timestamp)
if not thread:
raise UserError("El hilo no existe.")
return renderTemplate(template_filename, {"threads": [thread], "replythread": postid}, mobile)
def dynamicRead(parentid, ranges, mobile=False):
import re
board = Settings._.BOARD
if board['board_type'] != 1:
raise UserError("Esta sección no es un BBS y como tal no soporta lectura dinámica.")
# get entire thread
template_fname = "txt_thread.html"
thread = getThread(timestamp=parentid, mobile=mobile)
if not thread:
# Try the archive
import json
fname = Settings.ROOT_DIR + board["dir"] + "/kako/" + str(parentid) + ".json"
if os.path.isfile(fname):
import json
with open(fname) as f:
thread = json.load(f)
thread['posts'] = [dict(zip(thread['keys'], row)) for row in thread['posts']]
template_fname = "txt_archive.html"
else:
raise UserError('El hilo no existe.')
filtered_thread = {
"id": thread['id'],
"timestamp": thread['timestamp'],
"length": thread['length'],
"subject": thread['subject'],
"locked": thread['locked'],
"posts": [],
}
if 'size' in thread:
filtered_thread['size'] = thread['size']
no_op = False
if ranges.endswith('n'):
no_op = True
ranges = ranges[:-1]
# get thread length
total = thread["length"]
# compile regex
__multiple_ex = re.compile("^([0-9]*)-([0-9]*)$")
__single_ex = re.compile("^([0-9]+)$")
__last_ex = re.compile("^l([0-9]+)$")
start = 0
end = 0
# separate by commas (,)
for range in ranges.split(','):
# single post (#)
range_match = __single_ex.match(range)
if range_match:
postid = int(range_match.group(1))
if postid > 0 and postid <= total:
filtered_thread["posts"].append(thread["posts"][postid-1])
# go to next range
continue
# post range (#-#)
range_match = __multiple_ex.match(range)
if range_match:
start = int(range_match.group(1) or 1)
end = int(range_match.group(2) or total)
if start > total:
start = total
if end > total:
end = total
if start < end:
filtered_thread["posts"].extend(thread["posts"][start-1:end])
else:
list = thread["posts"][end-1:start]
list.reverse()
filtered_thread["posts"].extend(list)
# go to next range
continue
# last posts (l#)
range_match = __last_ex.match(range)
if range_match:
length = int(range_match.group(1))
start = total - length + 1
end = total
if start < 1:
start = 1
filtered_thread["posts"].extend(thread["posts"][start-1:])
continue
# calculate previous and next ranges
prevrange = None
nextrange = None
if __multiple_ex.match(ranges) or __last_ex.match(ranges):
if mobile:
range_n = 50
else:
range_n = 100
prev_start = start-range_n
prev_end = start-1
next_start = end+1
next_end = end+range_n
if prev_start < 1:
prev_start = 1
if next_end > total:
next_end = total
if start > 1:
prevrange = '%d-%d' % (prev_start, prev_end)
if end < total:
nextrange = '%d-%d' % (next_start, next_end)
if not no_op and start > 1 and end > 1:
filtered_thread["posts"].insert(0, thread["posts"][0])
if not filtered_thread["posts"]:
raise UserError("No hay posts que mostrar.")
post_preview = cut_msg(filtered_thread['posts'][0]['message'], Settings.HOME_LASTPOSTS_LENGTH)
return renderTemplate(template_fname, {"threads": [filtered_thread], "replythread": parentid, "prevrange": prevrange, "nextrange": nextrange, "preview": post_preview}, mobile, noindex=True)
def regenerateBoard(everything=False):
"""
Update front pages and every thread res HTML page
"""
board = Settings._.BOARD
op_posts = []
if everything:
op_posts = FetchAll("SELECT `id` FROM `posts` WHERE `boardid` = %s AND `parentid` = 0 AND IS_DELETED = 0" % board["id"])
# Use queues only if multithreading is enabled
if Settings.USE_MULTITHREADING:
request_queue = queue.Queue()
threads = [RegenerateThread(i, request_queue) for i in range(Settings.MAX_PROGRAM_THREADS)]
for t in threads:
t.start()
request_queue.put("front")
for post in op_posts:
request_queue.put(post["id"])
for i in range(Settings.MAX_PROGRAM_THREADS):
request_queue.put(None)
for t in threads:
t.join()
else:
regenerateFrontPages()
for post in op_posts:
regenerateThreadPage(post["id"])
def deletePost(postid, password, deltype=0, imageonly=False, quick=False):
"""
Remove post from database and unlink file (if present), along with all replies
if supplied post is a thread
"""
board = Settings._.BOARD
# make sure postid is numeric
postid = int(postid)
# get post
post = FetchOne("SELECT `id`, `timestamp`, `parentid`, `file`, `thumb`, `password`, `length` FROM `posts` WHERE `boardid` = %s AND `id` = %s LIMIT 1", (board["id"], postid))
# abort if the post doesn't exist
if not post:
raise UserError(_("There isn't a post with this ID. It was probably deleted."))
if password:
if password != post['password']:
raise UserError("No tienes permiso para eliminar este mensaje.")
elif not post["parentid"] and post["length"] >= Settings.DELETE_FORBID_LENGTH:
raise UserError("No puedes eliminar un hilo con tantas respuestas.")
elif (int(time.time()) - post["timestamp"]) > 86400:
raise UserError("No puedes eliminar un post tan viejo.")
# just update the DB if deleting only the image, otherwise delete whole post
if imageonly:
if post["file"]:
deleteFile(post)
UpdateDb("UPDATE `posts` SET `file` = '', `file_hex` = '', `thumb` = '', `thumb_width` = 0, `thumb_height` = 0 WHERE `boardid` = %s AND `id` = %s LIMIT 1", (board["id"], post['id']))
else:
if not post["parentid"]:
deleteReplies(post)
logging.info("Deleting post %d - parentid: %d" % (post["id"], post["parentid"]))
if deltype != 0 and post["parentid"]:
# Soft delete (recycle bin)
logging.info("Soft delete")
UpdateDb("UPDATE `posts` SET `IS_DELETED` = %s WHERE `boardid` = %s AND `id` = %s LIMIT 1", (deltype, board["id"], post["id"]))
else:
# Hard delete
logging.info("Hard delete")
if post["file"]:
deleteFile(post)
if post['parentid']:
numreplies = threadNumReplies(post["parentid"])
if numreplies > 2:
newlast = FetchOne('SELECT timestamp, email FROM posts WHERE boardid = %s AND parentid = %s AND timestamp < %s ORDER BY timestamp DESC LIMIT 1', (board['id'], post['parentid'], post['timestamp']))
else:
newlast = FetchOne('SELECT timestamp FROM posts WHERE boardid = %s AND id = %s LIMIT 1', (board['id'], post['parentid']))
UpdateDb("DELETE FROM `posts` WHERE `boardid` = %s AND `id` = %s LIMIT 1", (board["id"], post["id"]))
if post['parentid']:
UpdateDb("UPDATE `posts` SET last = %s, length = %s WHERE `id` = %s AND `boardid` = %s",
(newlast["timestamp"], threadNumReplies(post["parentid"]), post["parentid"], board["id"]))
if not post['parentid']:
if board['board_type'] == 1:
fname = Settings.ROOT_DIR + str(board["dir"]) + "/res/" + str(post["timestamp"]) + ".html"
else:
fname = Settings.ROOT_DIR + str(board["dir"]) + "/res/" + str(post["id"]) + ".html"
logging.info("Unlinking html " + fname)
try:
os.unlink(fname)
except FileNotFoundError:
logging.warn("Thread HTML (%s) didn't exist! Continuing..." % fname)
regenerateHome()
# rebuild thread and fronts if reply; rebuild only fronts if not
if post["parentid"]:
threadUpdated(post["parentid"])
else:
regenerateFrontPages()
def deleteReplies(thread):
board = Settings._.BOARD
# delete files first
replies = FetchAll("SELECT `parentid`, `file`, `thumb` FROM `posts` WHERE `boardid` = %s AND `parentid` = %s AND `file` != ''", (board["id"], thread["id"]))
for post in replies:
deleteFile(post)
# delete all replies from DB
UpdateDb("DELETE FROM `posts` WHERE `boardid` = %s AND `parentid` = %s", (board["id"], thread["id"]))
def deleteFile(post):
"""
Unlink file and thumb of supplied post
"""
board = Settings._.BOARD
try:
os.unlink(Settings.IMAGES_DIR + board["dir"] + "/src/" + post["file"])
except:
pass
# we don't want to delete mime thumbnails
if post["thumb"].startswith("mime"):
return
try:
os.unlink(Settings.IMAGES_DIR + board["dir"] + "/thumb/" + post["thumb"])
except:
pass
try:
os.unlink(Settings.IMAGES_DIR + board["dir"] + "/mobile/" + post["thumb"])
except:
pass
if int(post["parentid"]) == 0:
try:
os.unlink(Settings.IMAGES_DIR + board["dir"] + "/cat/" + post["thumb"])
except:
pass
def trimThreads():
"""
Delete any threads which have passed the MAX_THREADS setting
"""
board = Settings._.BOARD
logging.debug("Trimming threads (%s)" % board["dir"])
archived = False
trimmed = []
# Use limit of the board type
if board['board_type'] == 1:
limit = Settings.TXT_MAX_THREADS
else:
limit = Settings.MAX_THREADS
# trim expiring threads first
if board['maxage'] > 0:
t = time.time()
alert_time = int(round(board['maxage'] * Settings.MAX_AGE_ALERT))
time_limit = t + (alert_time * 86400)
old_ops = FetchAll("SELECT `id`, `timestamp`, `expires`, `expires_alert`, `length` FROM `posts` WHERE `boardid` = %s AND `parentid` = 0 AND IS_DELETED = 0 AND `expires` > 0 AND `expires` < %s LIMIT 50", (board['id'], time_limit))
for op in old_ops:
if t >= op['expires']:
# Trim old threads
logging.debug("Old OP 1: " + repr(op))
if board['archive'] and op["length"] >= Settings.ARCHIVE_MIN_LENGTH:
archiveThread(op["id"])
archived = True
deletePost(op["id"], None)
trimmed.append(op["id"])
else:
# Add alert to threads approaching deletion
UpdateDb("UPDATE `posts` SET expires_alert = 1 WHERE `boardid` = %s AND `id` = %s", (board['id'], op['id']))
# trim inactive threads next
if board['maxinactive'] > 0:
t = time.time()
oldest_last = t - (board['maxinactive'] * 86400)
old_ops = FetchAll("SELECT `id`, `length` FROM `posts` WHERE `boardid` = %s AND `parentid` = 0 AND IS_DELETED = 0 AND `last` < %s LIMIT 50", (board['id'], oldest_last))
logging.debug("Old OPs 2: " + repr(old_ops))
for op in old_ops:
if board['archive'] and op["length"] >= Settings.ARCHIVE_MIN_LENGTH:
archiveThread(op["id"])
archived = True
deletePost(op["id"], None)
trimmed.append(op["id"])
# select trim type by board
if board['board_type'] == 1:
trim_method = Settings.TXT_TRIM_METHOD
else:
trim_method = Settings.TRIM_METHOD
# select order by trim
if trim_method == 1:
order = 'last DESC'
elif trim_method == 2:
order = 'bumped DESC'
else:
order = 'timestamp DESC'
# Trim the last thread
op_posts = FetchAll("SELECT `id`, `length` FROM `posts` WHERE `boardid` = %s AND `parentid` = 0 AND IS_DELETED = 0 ORDER BY %s" % (board["id"], order))
if len(op_posts) > limit:
posts = op_posts[limit:]
logging.debug("Old OPs 3: " + repr(posts))
for post in posts:
if board['archive'] and op["length"] >= Settings.ARCHIVE_MIN_LENGTH:
archiveThread(post["id"])
archived = True
deletePost(post["id"], None)
trimmed.append(op["id"])
if archived:
regenerateKako()
logging.debug("Trimmed threads: " + repr(trimmed))
return trimmed
def autoclose_thread(parentid, t, replies):
"""
If the thread is crossing the reply limit, close it with a message.
"""
board = Settings._.BOARD
# decide the replylimit
if board['board_type'] == 1 and Settings.TXT_CLOSE_THREAD_ON_REPLIES > 0:
replylimit = Settings.TXT_CLOSE_THREAD_ON_REPLIES
elif Settings.CLOSE_THREAD_ON_REPLIES > 0:
replylimit = Settings.CLOSE_THREAD_ON_REPLIES
else:
return # do nothing
# close it if passing replylimit
#if replies >= replylimit or board["dir"] == "polka":
if replies >= replylimit:
notice_post = Post(board["id"])
notice_post["parentid"] = parentid
notice_post["name"] = "Sistema"
notice_post["message"] = "El hilo ha llegado al límite de respuestas.
Si quieres continuarlo, por favor crea otro."
notice_post["timestamp"] = t+1
notice_post["bumped"] = get_parent_post(parentid, board["id"])["bumped"]
notice_post["timestamp_formatted"] = str(replylimit) + " mensajes"
notice_post.insert()
UpdateDb("UPDATE `posts` SET `locked` = 1 WHERE `boardid` = %s AND `id` = %s LIMIT 1", (board["id"], parentid))
def pageNavigator(page_num, page_count, is_omitted=False):
"""
Create page navigator in the format of [0], [1], [2]...
"""
board = Settings._.BOARD
# No threads?
if page_count == 0:
return ''
# TODO nijigen HACK
first_str = "Primera página"
last_str = "Última página"
previous_str = _("Previous")
next_str = _("Next")
omitted_str = "Resto omitido"
pagenav = ""
if page_num == 0:
pagenav += first_str
else:
previous = str(page_num - 1)
if previous == "0":
previous = ""
else:
previous = previous + ".html"
pagenav += ''
pagenav += ""
for i in range(page_count):
if i == page_num:
pagenav += "[%d]" % i
else:
if i == 0:
pagenav += '[%d]' % (Settings.BOARDS_URL, board['dir'], i)
else:
pagenav += '[%d]' % (Settings.BOARDS_URL, board['dir'], i, i)
if i > 0 and (i % 10) == 0 and not is_omitted:
pagenav += '
'
elif i < 10:
pagenav += ' '
if is_omitted:
pagenav += "[" + omitted_str + "]"
pagenav += ""
pagenav += ""
next = (page_num + 1)
if next == page_count:
pagenav += last_str + ""
else:
pagenav += ''
return pagenav
def flood_check(t,post,boardid):
board = Settings._.BOARD
if not post["parentid"]:
maxtime = round(t - int(board['threadsecs']))
lastpost = FetchOne("SELECT `timestamp` FROM `posts` WHERE `ip` = INET6_ATON(%s) and `parentid` = 0 and `boardid` = %s and IS_DELETED = 0 AND timestamp > %s",
(post["ip"], boardid, maxtime))
# NO MATTER THE IP
if not lastpost:
lastpost = FetchOne("SELECT `timestamp` FROM `posts` WHERE `parentid` = 0 and `boardid` = %s and IS_DELETED = 0 AND timestamp > %s",
(boardid, maxtime))
else:
maxtime = round(t - int(board['postsecs']))
lastpost = FetchOne("SELECT `timestamp` FROM `posts` WHERE `ip` = INET6_ATON(%s) and `parentid` != 0 and `boardid` = %s and IS_DELETED = 0 AND timestamp > %s",
(post["ip"], boardid, maxtime))
if lastpost:
if post["parentid"]:
raise UserError(_("Flood detected. Please wait a moment before posting again."))
else:
lastpost = FetchOne("SELECT `timestamp` FROM `posts` WHERE `parentid`= 0 and `boardid`= %s and IS_DELETED = 0 ORDER BY `timestamp` DESC", (boardid,))
wait = int(int(board['threadsecs']) - (t - int(lastpost["timestamp"])))
raise UserError("Espera " + str(wait) + " segundos antes de crear otro hilo.")
def cut_msg(message, limit):
short_message = message.replace("
", " ")
short_message = short_message.split("
")[0]
short_message = re.compile(r"<[^>]*?>", re.DOTALL | re.IGNORECASE).sub("", short_message) # Removes HTML tags
if len(short_message) > limit:
if isinstance(short_message, str):
short_message = short_message[:limit] + "…"
short_message = re.compile(r"&(.(?!;))*$", re.DOTALL | re.IGNORECASE).sub("", short_message) # Removes incomplete HTML
return short_message
def getLastAge(board_type, limit):
threads = []
if board_type in [0, 1]:
sql = "SELECT posts.id, boards.name AS board_fulln, boards.subname AS board_name, board_type, boards.dir, timestamp, length, CASE WHEN posts.subject = boards.subject THEN posts.message ELSE posts.subject END AS content FROM posts INNER JOIN boards ON boardid = boards.id WHERE parentid = 0 AND IS_DELETED = 0 AND boards.secret = 0 AND posts.locked < 3 AND boards.board_type = %s ORDER BY bumped DESC LIMIT %s"
threads = FetchAll(sql, (board_type, limit) )
else:
sql = "SELECT posts.id, boards.name AS board_fulln, boards.subname AS board_name, board_type, boards.dir, timestamp, length, CASE WHEN posts.subject = boards.subject THEN posts.message ELSE posts.subject END AS content FROM posts INNER JOIN boards ON boardid = boards.id WHERE parentid = 0 AND IS_DELETED = 0 AND boards.secret = 0 AND posts.locked < 3 ORDER BY bumped DESC LIMIT %s"
threads = FetchAll(sql, (limit,) )
for post in threads:
post['content'] = cut_msg(post['content'], Settings.HOME_LASTPOSTS_LENGTH)
if post['board_type'] == 1:
post['url'] = '/%s/read/%d/l10' % (post['dir'], post['timestamp'])
else:
post['url'] = '/%s/res/%d.html' % (post['dir'], post['id'])
return threads
def getNewPosts(limit, rss=False):
sql = "SELECT posts.id, boards.name AS board_fulln, board_type, boards.dir, timestamp, posts.message FROM posts INNER JOIN boards ON posts.boardid = boards.id WHERE IS_DELETED = 0 AND boards.secret = 0 ORDER BY timestamp DESC LIMIT %s"
posts = FetchAll(sql, (limit,))
for post in posts:
post['content'] = cut_msg(post['message'], Settings.HOME_LASTPOSTS_LENGTH)
if rss:
post['timestamp_formatted'] = datetime.datetime.fromtimestamp(post['timestamp']).strftime('%Y-%m-%dT%H:%M:%S%Z')
if post['board_type'] == 1:
post['url'] = '/%s/read/%d/l10' % (post['dir'], post['timestamp'])
else:
post['url'] = '/%s/res/%d.html' % (post['dir'], post['id'])
return posts
def getNewThreads(limit):
threads = []
sql = "SELECT posts.id, boards.name AS board_fulln, boards.subname AS board_name, board_type, boards.dir, timestamp, CASE WHEN posts.subject = boards.subject THEN posts.message ELSE posts.subject END AS content FROM posts INNER JOIN boards ON boardid = boards.id WHERE parentid = 0 AND IS_DELETED = 0 AND boards.secret = 0 AND boards.id <> 13 AND posts.locked = 0 ORDER BY timestamp DESC LIMIT %s"
threads = FetchAll(sql, (limit,) )
for post in threads:
post['content'] = cut_msg(post['content'], Settings.HOME_LASTPOSTS_LENGTH)
if post['board_type'] == 1:
post['url'] = '/%s/read/%d' % (post['dir'], post['timestamp'])
else:
post['url'] = '/%s/res/%d.html' % (post['dir'], post['id'])
return threads
def regenerateHome():
"""
Update index.html in the boards directory with useful data for users
"""
logging.debug("Updating home")
t = datetime.datetime.now()
limit = Settings.HOME_LASTPOSTS
last_bbs = getLastAge(1, limit)
last_ib = getLastAge(0, limit)
template_values = {
'header': Settings.SITE_TITLE,
'slogan': Settings.SITE_SLOGAN,
'latest_news': FetchAll("SELECT `timestamp`, `message`, `timestamp_formatted` FROM `news` WHERE `type` = '2' ORDER BY `timestamp` DESC LIMIT %s", (Settings.HOME_NEWS,) ),
'latest_age_bbs': last_bbs,
'latest_age_ib': last_ib,
'latest_age_num': limit,
'new_threads': getNewThreads(Settings.HOME_NEWTHREADS),
'rss_enabled': Settings.ENABLE_RSS,
}
page_rendered = renderTemplate('home.html', template_values)
f = open(Settings.HOME_DIR + "home.html", "w")
try:
f.write(page_rendered)
finally:
f.close()
if Settings.ENABLE_RSS:
rss_rendered = renderTemplate('home.rss', {'posts': getNewPosts(10, True)} )
f = open(Settings.HOME_DIR + "bai.rss", "w")
try:
f.write(rss_rendered)
finally:
f.close()
def regenerateNews():
"""
Update news.html in the boards directory with older news
"""
posts = FetchAll("SELECT * FROM `news` WHERE `type` = '1' ORDER BY `timestamp` DESC")
template_values = {
'title': 'Noticias',
'posts': posts,
'header': Settings.SITE_TITLE,
'slogan': Settings.SITE_SLOGAN,
'navbar': False,
}
page_rendered = renderTemplate('news.html', template_values)
f = open(Settings.HOME_DIR + "noticias.html", "w")
try:
f.write(page_rendered)
finally:
f.close()
def regenerateTrash():
posts = FetchAll("SELECT posts.id AS id, timestamp_formatted, posts.message AS message, boards.name AS board FROM `posts` INNER JOIN `boards` ON boardid = boards.id WHERE IS_DELETED = 2 AND timestamp > %s ORDER BY `timestamp` DESC", (int(time.time()) - 31557600*3,) )
for i in range(len(posts)):
posts[i]["timestamp_formatted"] = posts[i]["timestamp_formatted"].split()[0]
posts[i]["message"] = formatting.remove_html(posts[i]["message"])
template_values = {'posts': posts}
page_rendered = renderTemplate('trash.html', template_values)
with open(Settings.HOME_DIR + "basural.html", "w") as f:
f.write(page_rendered)
def regenerateAccess():
if not Settings.HTACCESS_GEN:
return False
boards = FetchAll('SELECT `dir` FROM `boards`')
global_boards = [board['dir'] for board in boards if board['dir'] not in Settings.EXCLUDE_GLOBAL_BANS]
bans = []
global_bans = []
board_bans = {}
if Settings.ENABLE_BANS:
bans = FetchAll("SELECT `ipstr`, `boards` FROM `bans` WHERE `blind` = 1 ORDER BY `ipstart` ASC")
for ban in bans:
if ban["boards"]:
boards = str2boards(ban["boards"])
for board in boards:
board_bans.setdefault(board, []).append(ban["ipstr"])
else:
global_bans.append(ban["ipstr"])
# Generate .htaccess for each board
""" TODO: htaccess
for board in listbans.keys():
template_values = {
'ips': listbans[board],
'dir': board,
}
page_rendered = renderTemplate('htaccess', template_values)
f = open(Settings.ROOT_DIR + board + "/.htaccess", "w")
try:
f.write(page_rendered)
finally:
f.close()"""
template_values = {
'global_boards': global_boards,
'global_bans': global_bans,
'board_bans': board_bans,
}
page_rendered = renderTemplate('bans_geo', template_values)
with open(Settings.ROOT_DIR + "/bans_geo.conf", "w") as f:
f.write(page_rendered)
page_rendered = renderTemplate('bans_locations', template_values)
with open(Settings.ROOT_DIR + "/bans_locations.conf", "w") as f:
f.write(page_rendered)
os.utime(Settings.ROOT_DIR + "/bans.conf", None)
return True
def regenerateKako():
board = Settings._.BOARD
threads = FetchAll("SELECT * FROM archive WHERE boardid = %s ORDER BY timestamp DESC", (board['id'],) )
page = renderTemplate('kako.html', {'threads': threads})
with open(Settings.ROOT_DIR + board["dir"] + "/kako/index.html", "w") as f:
f.write(page)
def make_url(postid, post, parent_post, noko, mobile):
board = Settings._.BOARD
parentid = post["parentid"]
if not parentid:
parentid = postid
if mobile:
if not noko:
url = Settings.CGI_URL + 'mobile/' + board["dir"]
elif board["board_type"] == 1:
url = "%s/mobileread/%s/%s/l10#form" % (Settings.CGI_URL, board["dir"], parent_post['timestamp'])
else:
url = "%s/mobileread/%s/%s#%s" % (Settings.CGI_URL, board["dir"], parentid, postid)
else:
if not noko:
url = Settings.BOARDS_URL + board["dir"] + "/"
elif board["board_type"] == 1:
url = "%s/read/%s/l50#bottom" % (Settings.BOARDS_URL + board["dir"], str(parent_post['timestamp']))
else:
url = "%s/res/%s.html#%s" % (Settings.BOARDS_URL + board["dir"], str(parentid), postid)
return url
def make_redirect(url, timetaken=None):
board = Settings._.BOARD
randomPhrase = getRandomLine('quotes.conf')
return renderTemplate('redirect.html', {'url': url, 'message': randomPhrase, 'timetaken': timetaken})
def archiveThread(postid):
import json
board = Settings._.BOARD
thread = getThread(postid, False)
post_preview = cut_msg(thread['posts'][0]['message'], Settings.HOME_LASTPOSTS_LENGTH)
page = renderTemplate("txt_archive.html", {"threads": [thread], "replythread": postid, "preview": post_preview}, False)
with open(Settings.ROOT_DIR + board["dir"] + "/kako/" + str(thread['timestamp']) + ".html", "w") as f:
f.write(page)
thread['keys'] = ['num', 'IS_DELETED', 'name', 'tripcode', 'email', 'message', 'timestamp_formatted']
thread['posts'] = [[row[key] for key in thread['keys']] for row in thread['posts']]
try:
with open(Settings.ROOT_DIR + board["dir"] + "/kako/" + str(thread['timestamp']) + ".json", "w") as f:
json.dump(thread, f, indent=0)
except:
raise UserError("Can't archive: %s" % thread['timestamp'])
UpdateDb("REPLACE INTO archive (oldid, boardid, timestamp, subject, length) VALUES (%s, %s, %s, %s, %s)", (thread['id'], board['id'], thread['timestamp'], thread['subject'], thread['length']))
def discord_hook(post, parent_post, url):
if not Settings.DISCORD_HOOK_URL:
return
import urllib.request, urllib.error, urllib.parse
import json
board = Settings._.BOARD
if parent_post:
ptitle = "Nueva respuesta en {} ({})".format(html.unescape(parent_post["subject"]), parent_post["length"]+1)
else:
ptitle = "Nuevo hilo: {}".format(html.unescape(post['subject']))
data = {"embeds": [{
"type": "rich",
"title": ptitle,
"description": html.unescape(cut_msg(post['message'], 200)),
"color": 11910504,
"url": "https://bienvenidoainternet.org" + url, # TODO: Parametrizar.
"timestamp": datetime.datetime.utcfromtimestamp(post['timestamp']).isoformat(),
"footer": { "text": board['name'] },
}]
}
if post['thumb']:
data["embeds"][0]["thumbnail"] = {"url": "%s%s/thumb/%s" % (Settings.HOME_URL, board['dir'], post['thumb'])}
jsondata = json.dumps(data, separators=(',',':')).encode('utf-8')
try:
req = urllib.request.Request(Settings.DISCORD_HOOK_URL)
req.add_header('Content-Type', 'application/json')
req.add_header('Content-Length', len(jsondata))
req.add_header('User-Agent', "weabot/0.1")
response = urllib.request.urlopen(req, jsondata)
except urllib.error.HTTPError as e:
raise Exception(e.read())