root/trac/hacks/marketplugin/0.9/tracmarket/web_ui.py

import random
import re
import sys
import urllib

from trac.core import *
from trac.web.api import IRequestPostProcessor
from trac.web.chrome import INavigationContributor, ITemplateProvider, add_stylesheet
from trac.web.main import IRequestHandler
from trac.wiki.model import WikiPage
from trac.wiki.api import IWikiWorkflowController
from trac.util import escape, format_datetime, pretty_timedelta, Markup
from trac.wiki.formatter import wiki_to_html, wiki_to_oneliner

from api import *
from mq import *

class Market(Component):
    implements(INavigationContributor, ITemplateProvider,
            IRequestHandler, IRequestPostProcessor)

    controllers = ExtensionPoint(IMarketController)

    # INavigationContributor methods
    def get_active_navigation_item(self, req):
        return 'market'
                
    def get_navigation_items(self, req):
        yield 'mainnav', 'market', Markup('<a href="%s">Market</a>' \
                                  % self.env.href.market())

    # ITemplateProvider methods
    def get_templates_dirs(self):
        """
        Return the absolute path of the directory containing the provided
        ClearSilver templates.
        """
        from pkg_resources import resource_filename
        return [resource_filename(__name__, 'templates')]

    def get_htdocs_dirs(self):
        """
        Return a list of directories with static resources (such as style
        sheets, images, etc.)

        Each item in the list must be a `(prefix, abspath)` tuple. The
        `prefix` part defines the path in the URL that requests to these
        resources are prefixed with.
        
        The `abspath` is the absolute path to the directory containing the
        resources on the local file system.
        """
        from pkg_resources import resource_filename
        return [('tracmarket', resource_filename(__name__, 'htdocs'))]

    # IRequestHandler methods
    def match_request(self, req):
        return re.match('/market(/|$)', req.path_info)
    
    def process_request(self, req):
        action = req.args.get('action')

        # req.args['hdfdump'] = 1

        add_stylesheet(req, 'common/css/trac.css')
        # add_stylesheet(req, 'common/css/wiki.css')
        add_stylesheet(req, 'tracmarket/css/market.css')
        # add_stylesheet(req, 'tracmarket/css/luna/luna.css')
        template = 'market.cs'
        mime_type = 'text/html'

        # construct cmd
        cmds = []
        if action == 'order':
            symbol = req.args.get('symbol')
            side = req.args.get('side')
            size = req.args.get('size')
            price = req.args.get('price')
            # comment = req.args.get('comment')
            cmd = "%s %s %s %s" % (symbol, side, size, price)
            cmds.append(cmd)
            cmds.append(symbol)
        elif req.args.get('commands'):
            script = req.args.get('commands')
            script = urllib.unquote(script)
            for cmd in script.split('\n'):
                cmd = cmd.strip()
                if not cmd:
                    continue
                if cmd.startswith('#'):
                    continue
                cmds.append(cmd)
        else:
            m = re.match('/market/(.*)', req.path_info)
            if m and m.group(1):
                cmd = m.group(1)
                # reformat cmd from "wiki.FooPage.3?b+50+.23+TMP"
                #                to "wiki.FooPage.3 b 50 .23 TMP"
                parts = cmd.split('?', 2)
                symbol = parts[0]
                if len(parts) > 1:
                    args = parts[1]
                    args = urllib.unquote_plus(args)
                    cmd = "%s %s" % (symbol, args)
                else:
                    cmd = symbol
                cmds.append(cmd)
        if not cmds:
            cmds = ['portfolio', 'traders show_worth']
        
        # create bus
        ctl = list(self.controllers)[0]
        bus = ctl.open(req=req)

        # set up initial subscriptions
        bus.subscribe('debug', reader=self.debug)
        bus.subscribe('res.', reader=self.build_response)

        # send cmds
        self.env.log.debug("commands " + '\n'.join(cmds))
        try:
            for cmd in cmds:
                self.env.log.debug("XA")
                msg = Msg(cmd)
                bus.send('market.cmd', msg)
                self.env.log.debug("XB")
            msg = Msg('end')
        except InsufficientFunds, e:
            self.env.log.debug("XC")
            ctl.rollback(bus)
            bus.ctx.res.error(str(e)) 
        except:
            self.env.log.debug("XD")
            ctl.rollback(bus)
            raise
        else:
            self.env.log.debug("XE")
            ctl.commit(bus)
            self.env.log.debug("XF")

        # XXX we assume the above send() calls are always blocking
        if req.args.get('resdump'):
            req.hdf['market.resdump'] = bus.ctx.res.dump()
        self.env.log.debug("G")
        return template,mime_type

    def build_response(self, bus, group, msg):
        self.env.log.debug("H")
        req = bus.ctx.req
        group = re.sub('^res\.', '', msg.group, count=1)
        req.hdf[group] = msg.data

    def debug(self, bus, group, msg):
        self.env.log.debug("%s: %s" % (group, msg.data))

    # IRequestPostProcessor methods

    def process(self, req, template, mime_type):
        page_name = req.hdf.get('wiki.page_name')
        version = req.hdf.get('wiki.version') 
        action = req.hdf.get('wiki.action') 
        mime_type = mime_type or 'text/html'
        if page_name and version and action == 'view' \
                and template == 'wiki.cs' and mime_type == 'text/html':
            # override the default wiki template
            template = 'market/wiki_override.cs'
            # add the publishing cost calculations to the hdf
            cmd = "wiki.%s estimate" % (page_name)
            ctl = list(self.controllers)[0]
            ctx = ctl.run_cmd(req=req, cmd=cmd) 
            ctx.res.to_hdf(req.hdf)
        return template, mime_type