source: trollforge/dsp-trollforge/MovableType/Mtpost.py @ 212 Tweet

#!/usr/bin/env python
#
# mt-post.py:
#   Post a comment on any MT blog entry. Also defeats stupid SCode "security"
#   check that is so annoying when using a text browser.
#
# Usage:
#   mt-post.py "blog_entry_url" "nick" "valid@email.com" "multi\nline\ncomment"
#
# Copyright: (c) 2004 Sam Hocevar <sam@zoy.org>
#   This program is free software; you can redistribute it and/or
#   modify it under the terms of the Do What The Fuck You Want To
#   Public License as published by Banlu Kemiyatorn. See
#   http://sam.zoy.org/projects/COPYING.WTFPL for more details.
#

import gd, urllib, urllib2, sys, re

# Behave nicely, like MSIE
user_agent = "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1; SV1; .NET CLR 1.1.4322)"

# Guess a glyph from its bounding box in the picture. This function is very,
# very naive. But since mt-scode is very, very naive as well, it's more than
# enough.
def guess_glyph(im, xmin, xmax, ymin, ymax):
    values = [162, 131, 150, 139, 155, 159, 181, 90, 180, 170]
    count = 0
    for x in range(xmin, xmax):
        for y in range(ymin, ymax):
            count += im.getPixel((x,y)) * (y - ymin)
    if count not in values:
        return "?"
    return values.index(count)

# Remove background stuff from an scode picture
def remove_background(im):
    (w,h) = im.size()
    # Stats on first 3 lines to determine background colours
    stats = [0, 0, 0, 0]
    fg = 0
    bg = 0
    for y in range(3):
        for x in range(w):
            p = im.getPixel((x,y))
            if p < 4:
                stats[p] += 1
    for i in range(1,4):
        if stats[i] < stats[fg]:
            fg = i
        if stats[i] > stats[bg]:
            bg = i
    # Unify background colours
    for y in range(h):
        for x in range(w):
            p = im.getPixel((x,y))
            if p == fg:
                im.setPixel((x,y),1)
            else:
                im.setPixel((x,y),0)

# Decode an scode picture from its URL
def sdecode(url):
    r = urllib2.Request(url)
    r.add_header('User-Agent', user_agent)
    f = urllib2.build_opener().open(r)
    im = gd.image(f, "png")
    (w,h) = im.size()
    f.close()
    remove_background(im)
    # Find ymin/ymax bounding box
    ymin = -1
    ymax = -1
    for y in range(h):
        found = 0
        for x in range(w):
            p = im.getPixel((x,y))
            if p == 1:
                found = 1
                break
        if found:
            if ymin == -1:
                ymin = y
            else:
                ymax = y + 1
    # Read character cells
    result = ""
    incell = 0
    for x in range(w):
        found = 0
        for y in range(ymin, ymax):
            p = im.getPixel((x,y))
            if p == 1:
                found = 1
                break
        if found and not incell:
            incell = 1
            xmin = x
        if not found and incell:
            incell = 0
            result += str(guess_glyph(im, xmin, x, ymin, ymax))
    return result

def mt_post(url, nick, email, comment):
    data = {}
    re_scode = re.compile('.*src="([^"]*mt-scode[^"]*)".*\n')
    re_action = re.compile('.*action="([^"]*mt-comments[^"]*)".*\n')
    re_hidden = re.compile('.*type="hidden".*name="([^"]*)".*value="([^"]*)".*\n')
    r = urllib2.Request(url)
    r.add_header('User-Agent', user_agent)
    f = urllib2.build_opener().open(r)
    while True:
        l = f.readline()
        if not l:
            break
        # Detect an scode token
        m = re_scode.match(l)
        if m:
            scode = sdecode(m.group(1))
            data['scode'] = scode
        # Detect a hidden input
        m = re_hidden.match(l)
        if m:
            data[m.group(1)] = m.group(2)
        # Find the form target
        m = re_action.match(l)
        if m:
            target = m.group(1)
    f.close()
    data['author'] = nick
    data['email'] = email
    data['text'] = comment
    data['post'] = "yes"
    r = urllib2.Request(target, urllib.urlencode(data))
    r.add_header('User-Agent', user_agent)
    f = urllib2.build_opener().open(r)
    while True:
        l = f.readline()
        if not l:
            break
    f.close()

mt_post(sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4])