hello-ManaMarket-tmwa/tradey.py

#!/usr/bin/python
"""
    Copyright 2011, Dipesh Amin <yaypunkrock@gmail.com>
    Copyright 2011, Stefan Beller <stefanbeller@googlemail.com>

    This file is part of tradey, a trading bot in the mana world
    see www.themanaworld.org
"""
import time
import os
import xml.dom.minidom
from subprocess import call
from xml.etree.ElementTree import *

def clean_xml(parse):
    data = ''
    pos_start = 0
    while parse.find('<', pos_start) != -1:
        pos_start = parse.find('<', pos_start)
        pos_end = parse.find('>', pos_start+1)
        data += parse[pos_start:pos_end+1]
        pos_start = pos_end
    return data

class UserTree:
    def __init__(self):
        self.tree = ElementTree(file="data/user.xml")
        self.root = self.tree.getroot()

    def add_user(self, name, stalls, al):
        if self.get_user(name) == -10:
            user = SubElement(self.root, "user")
            user.set("name", name)
            user.set("stalls", str(stalls))
            user.set("used_stalls", str(0))
            user.set("money", str(0))
            user.set("id", str(0))
            user.set("accesslevel", str(al))
            self.save()

    def get_user(self, name):
        for elem in self.root:
            if elem.get("name") == name:
                return elem
        return -10

    def remove_user(self, name):
        for elem in self.root:
            if elem.get("name") == name:
                self.root.remove(elem)
                self.save()
                return 1
        return -10

    def save(self):
        # Be sure to call save() after any changes to the tree.
        f = open('data/user.xml', 'w')
        dom = xml.dom.minidom.parseString(clean_xml(tostring(self.root)))
        f.write(dom.toprettyxml('    '))
        f.close()

class ItemTree:
    def __init__(self):
        self.save_file = 'data/sale.xml'
        self.tree = ElementTree(file=self.save_file)
        self.root = self.tree.getroot()
        self.u_id = set()
        self.id_itter = 1

        for elem in self.root:
            self.u_id.add(int(elem.get("uid")))

    def getId(self):
        while self.id_itter in self.u_id:
                self.id_itter += 1
        self.u_id.add(self.id_itter)
        return self.id_itter

    def remove_id(self, uid):
        # Free up used id's.
        self.u_id.remove(uid)

    def add_item(self, name, item_id, amount, price):
        user = SubElement(self.root, "item")
        user.set("name", name)
        user.set("itemId", str(item_id))
        user.set("price", str(price))
        user.set("add_time", str(time.time()))
        user.set("amount", str(amount))
        user.set("uid", str(self.getId()))
        self.save()

    def get_uid(self, uid):
        for elem in self.root:
            if elem.get("uid") == str(uid):
                return elem
        return -10

    def remove_item_uid(self, uid):
        for elem in self.root:
            if elem.get("uid") == str(uid):
                self.root.remove(elem)
                self.remove_id(uid)
                self.save()
                return 1
        return -10

    def save(self):
        # Be sure to call save() after any changes to the tree.
        f = open(self.save_file, 'w')
        dom = xml.dom.minidom.parseString(clean_xml(tostring(self.root)))
        f.write(dom.toprettyxml('    '))
        f.close()

class StackTree(ItemTree):
    def __init__(self):
        self.save_file = 'data/stack.xml'
        self.tree = ElementTree(file=self.save_file)
        self.root = self.tree.getroot()
        self.u_id = set()
        self.id_itter = 101
        self.next_id = self.id_itter

        for elem in self.root:
            self.u_id.add(int(elem.get("uid")))

    def add_item(self, name, item_id, amount, price):
        user = SubElement(self.root, "item")
        user.set("name", name)
        user.set("itemId", str(item_id))
        user.set("price", str(price))
        user.set("amount", str(amount))
        user.set("uid", str(self.getId()))
        self.save()

    # Here we take the oldest id put into stack.xml so we have FIFO
    def get_next_id(self):
        if len(self.u_id) == 0:
            self.next_id = self.id_itter
            return
        else:
            # TODO Here I'm handling rotation (gets a high number then
            # back to 101). Any better manners to do this?
            if min(self.u_id) - self.next_id < 0:
               next_uid = self.next_id - min(self.u_id)
               self.next_id = self.id_itter + next_uid + 1
            else:
               next_uid = min(self.u_id) - self.next_id
               self.next_id += next_uid
            return
        

    def remove_item_uid(self, uid):
        for elem in self.root:
            if elem.get("uid") == str(uid):
                self.root.remove(elem)
                self.remove_id(uid)
                self.save()
                return 1
        self.get_next_id()
        return -10

class DelistedTree(ItemTree):
    def __init__(self):
        self.save_file = 'data/delisted.xml'
        self.tree = ElementTree(file=self.save_file)
        self.root = self.tree.getroot()
        self.u_id = set()
        self.id_itter = 301

        for elem in self.root:
            self.u_id.add(int(elem.get("uid")))

    def add_item(self, name, item_id, amount):
        user = SubElement(self.root, "item")
        user.set("name", name)
        user.set("itemId", str(item_id))
        user.set("amount", str(amount))
        user.set("uid", str(self.getId()))
        self.save()

def saveData(commitmessage = "commit"):
    # This assumes the current working directory is the tradey directory.
    os.chdir("data")
    call(["git", "commit","-a", '-m "' + commitmessage + '"'])
    os.chdir("..")

if __name__ == '__main__':
    print "Do not run this file directly. Run main.py"