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

import os
import re
import sys
import time
import types
import urllib

from trac.core import *
from trac.env import IEnvironmentSetupParticipant

from tracmarket.api import *
from tracmarket.command import *

SequenceType = (types.ListType, types.TupleType, types.GeneratorType)

class Traders(Component):
    implements(IMessageBusSubscriber)

    def __init__(self):
        self.exchange_authname = self.config.get(
            'market', 'exchange_authname', 'tracmarket')
        self.currency = self.config.get('market', 'currency', 'TMP')

    # IMessageBusSubscriber

    def solicit(self, bus):
        bus.subscribe('market.cmd.traders', reader=self.recv_cmd)

    def recv_cmd(self, bus, group, msg):
        verb = msg.verb
        args = msg.args
        func = self.parse_verb(self, verb)
        if args:
            func(bus, *args)
        else:
            func(bus)

    def parse_verb(self, interpreter, verb):
        method = "cmd_%s" % verb
        if not hasattr(interpreter, method):
            raise SyntaxError, "unknown verb: %s" % ( verb)
        return getattr(interpreter, method)

    # internal

    def cmd_show_worth(self, bus, order='desc', limit=5):
        # traders show_worth top 5
        def by_desc_worth(x, y):
            return cmp(worth[x] > worth[y])

        # XXX quick hack; refactor to use bus to get this info
        ctx = bus.ctx
        db = ctx.db
        ledger = ctx.ledger
        worth = {}
        balances = ledger.balances(db=db, symbol=self.currency)
        for entity,account,symbol,balance in balances:
            if entity in ('anonymous', self.exchange_authname):
                continue
            worth.setdefault(entity, 0)
            acct = ledger.getacct(db=db, entity=entity, account=account)
            type = acct.type
            if type == 'asset':
                worth[entity] += balance
            if type == 'liability':
                worth[entity] -= balance
        # XXX ignoring both order and limit
        keys = worth.keys()
        keys.sort(lambda x,y: cmp(worth[y], worth[x]))
        net_worth = []
        for entity in keys:
            net_worth.append(dict(entity=entity, worth=worth[entity]))
        # msg = Msg(net_worth)
        # XXX use bus
        ctx.res.market.traders.net_worth = net_worth