Mercurial > mamba

"""Where mamba's hang out with each other."""

from werkzeug.utils import secure_filename
from flask import Flask, request, abort

from datetime import datetime
from ConfigParser import SafeConfigParser
import xmlrpclib
import time
import os
import sys
import socket
import json

app = Flask(__name__)


def path(ctype):
    main = app.config.forest.main
    if ctype == "curated":
        return os.path.join(main.level_folder, 'curated')
    elif ctype == "uncurated":
        return os.path.join(main.level_folder, 'uncurated')
    abort(404, "Not found")


def list_levels(folder):
    endl = len(".txt")
    files = [x[:-endl] for x in os.listdir(folder)
             if not x.startswith('.') and x.endswith('.txt')]
    return "\n".join(files)


@app.route("/<ctype>/index")
def index(ctype):
    ctype = path(ctype)
    return list_levels(ctype)


@app.route("/<ctype>/level/<levelname>")
def level(ctype, levelname):
    ctype = path(ctype)
    levelname = "%s.txt" % secure_filename(levelname)
    levelpath = os.path.join(ctype, levelname)
    if not os.path.isfile(levelpath):
        abort(404, "Level not found. Hsss.")
    with open(levelpath) as level:
        return level.read()


@app.route("/save/<levelname>", methods=['GET', 'POST'])
def save(levelname):
    ts = datetime.now().strftime("%Y%m%d.%H%M%S")
    levelname = "%s.%s.txt" % (secure_filename(levelname), ts)
    levelpath = os.path.join(path("uncurated"), levelname)
    if request.method == 'POST':
        if os.path.exists(levelpath):
            abort(409, "Mamba already resident.")
        leveldata = request.form['data'].encode('ascii')
        with open(levelpath, 'w') as level:
            level.write(leveldata)
        inform_cia(levelname, "New level uploaded.", branch="uncurated")
        inform_irker(levelname, "New level uploaded.", branch="uncurated")
        return "Ssss."
    else:
        abort(405, "Post levels here. Hsss.")


MAMBA_VERSION = "0.1"
MAMBA_URL = "https://ctpug.org.za/hg/mamba"
IRKER_PORT = 6659
CIA_URL = "http://cia.navi.cx"
CIA_MSG_TEMPLATE = """
<message
 xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 xsi:noNamespaceSchemaLocation="schema.xsd">
   <generator>
       <name>Mutable Mamba Level Server</name>
       <version>%(mamba_version)s</version>
       <url>%(mamba_url)s</url>
   </generator>
   <source>
       <project>%(project)s</project>
       <module>%(module)s</module>
       <branch>%(branch)s</branch>
   </source>
   <timestamp>
       %(timestamp)d
   </timestamp>
   <body>
       <commit>
           <revision>%(revision)s</revision>
           <author>%(author)s</author>
           <files>
               <file>%(file)s</file>
           </files>
           <log>
               %(log)s
           </log>
       </commit>
   </body>
</message>
"""


def inform_cia(filename, log, branch='uncurated'):
    if app.config.forest.cia is None:
        return
    cia = app.config.forest.cia
    msg = CIA_MSG_TEMPLATE % {
        'mamba_version': MAMBA_VERSION,
        'mamba_url': MAMBA_URL,
        'project': cia.project,
        'module': 'level-server',
        'branch': branch,
        'timestamp': int(time.time()),
        'revision': '0',
        'author': 'unknown',
        'file': filename,
        'log': log,
        }
    srv = xmlrpclib.Server(cia.url)
    srv.hub.deliver(msg)


def format_irker_message(project, filename, log, branch='uncurated'):
    return "%(project)s: %(branch)s * %(filename)s: %(log)s" % {
        "project": project,
        "filename": filename,
        "log": log,
        "branch": branch,
    }


def inform_irker(filename, log, branch='uncurated'):
    if app.config.forest.irker is None:
        return
    irker = app.config.forest.irker
    privmsg = format_irker_message(irker.project, filename, log, branch)
    message = json.dumps({"to": irker.channels, "privmsg": privmsg})
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.connect(irker.host, irker.port)
        sock.sendall(message + "\n")
    finally:
        sock.close()


class ForestConfig(object):

    class SubConfig(object):
        """Holder for sub-config."""

    def __init__(self, filenames):
        self._config = SafeConfigParser()
        self._config.read(filenames)
        self.main = self._parse("main")
        self.irker = self._parse("irker")
        self.cia = self._parse("cia")

    def _parse(self, section):
        if self._config.has_section(section):
            conf = dict(self._config.items(section))
        else:
            conf = {}
        return getattr(self, "_parse_%s" % section)(conf)

    def _parse_main(self, conf):
        main = self.SubConfig()
        main.host = conf.get('host', '0.0.0.0')
        main.port = int(conf['port'])
        main.level_folder = conf['level_folder']
        return main

    def _parse_irker(self, conf):
        if 'project' not in conf:
            return None
        irker = self.SubConfig()
        irker.project = conf['project']
        irker.host = conf.get('host', 'localhost')
        irker.port = conf.get('port', IRKER_PORT)
        irker.channels = conf['channels']
        return irker

    def _parse_cia(self, conf):
        if 'project' not in conf:
            return None
        cia = self.SubConfig()
        cia.project = conf['project']
        cia.url = conf.get('url', CIA_URL)
        return cia


USAGE = """Usage: python -m mamba.forest <config file>

The config file should look like:

[main]
host=0.0.0.0  # optional
port=8000
level_folder=./my/levels

[irker]
host=localhost  # optional
port=8001
project=mamba-levels
channels=channel1,channel2
"""


if __name__ == "__main__":
    if len(sys.argv) != 2:
        print USAGE
        sys.exit(1)

    app.config.forest = ForestConfig([sys.argv[1]])
    main = app.config.forest.main

    for ctype in ("curated", "uncurated"):
        folder = os.path.join(main.level_folder, ctype)
        if not os.path.exists(folder):
            os.makedirs(folder)
        if not os.path.isdir(folder):
            print "Level folder must be a folder."
            sys.exit(1)
    cia_project = sys.argv[3] if len(sys.argv) > 3 else None

    # app.debug = True
    app.run(host=main.host, port=main.port)
author	Simon Cross <hodgestar@gmail.com>
date	Tue, 20 Nov 2012 23:18:46 +0200
parents	16344424dfcc
children	c296d10bd5f9