Changeset 77 for pycgibattler

Timestamp:

12/24/10 02:16:07 (14 years ago)

Author:

atzm

Message:

api support, for experimental test

File:

• pycgibattler/trunk/index.cgi (modified) (4 diffs)

pycgibattler/trunk/index.cgi

@@ -95 +95 @@
             return code, warrior
 
-    def flatten(self):
+    def entry(self):
         code, warrior = self.load()
         return {
@@ -103 +103 @@
             'warrior': warrior,
             'skills':  [warrior.skill(s) for s in warrior.skill_list()],
+        }
+
+    def flatten(self):
+        code, warrior = self.load()
+        skills = []
+        for name in warrior.skill_list():
+            sk = warrior.skill(name)
+            skills.append({
+                'name':  name,
+                'type':  str(sk.skilltype()).split('.')[-1],
+                'level': sk.level(),
+                'point': sk.point(),
+            })
+        return {
+            'id':            self.id(),
+            'mtime':         int(self.mtime()),
+            'name':          warrior.name(),
+            'hitpoint':      warrior.hitpoint(),
+            'skillpoint':    warrior.skillpoint(),
+            'strength':      warrior.strength(),
+            'concentration': warrior.concentration(),
+            'defense':       warrior.defense(),
+            'agility':       warrior.agility(),
+            'luck':          warrior.luck(),
+            'skills':        skills,
         }
 
@@ -216 +241 @@
 
 
+def jsondump(data):
+    sys.stdout.write('Content-type: application/json; charset=UTF-8\r\n\r\n')
+    sys.stdout.write(json.dumps(data, ensure_ascii=False) + '\n')
+    sys.stdout.flush()
+
+
 def main():
     form = cgi.FieldStorage()
 
+    if 'mode' in form and form.getfirst('mode') == 'json':
+        if 'id' in form:
+            data = {}
+            ids = form['id']
+
+            if type(ids) is not list:
+                ids = [ids]
+
+            if len(ids) > CONFIG.getint('battle', 'max_entries'):
+                return jsondump({})
+
+            try:
+                for id_ in ids:
+                    data[id_.value] = CharacterManager(id_.value).flatten()
+            except:
+                return jsondump({})
+
+            return jsondump(data)
+
+        return jsondump([cm.id() for cm in CharacterManager.list()])
+
     if 'id' in form:
-        entries = [CharacterManager(form.getfirst('id')).flatten()]
+        entries = [CharacterManager(form.getfirst('id')).entry()]
         return httpdump(CONFIG.get('template', 'character'), entries=entries)
 
     if 'warrior' in form:
         ids = form['warrior']
-        if type(ids) is list:
-            if len(ids) > CONFIG.getint('battle', 'max_entries'):
-                raise ValueError('battle warriors too long')
-            warriors = [CharacterManager(id_.value).load()[1] for id_ in ids]
-            result = do_battle(warriors)
-            return httpdump(CONFIG.get('template', 'battle'), result=result)
+        if type(ids) is not list:
+            ids = [ids]
+        if len(ids) > CONFIG.getint('battle', 'max_entries'):
+            raise ValueError('battle warriors too long')
+        warriors = [CharacterManager(id_.value).load()[1] for id_ in ids]
+        result = do_battle(warriors)
+        return httpdump(CONFIG.get('template', 'battle'), result=result)
 
     if 'filename' in form:
@@ -241 +294 @@
     CharacterManager.sweep()
 
-    entries = [cm.flatten() for cm in CharacterManager.list()]
+    entries = [cm.entry() for cm in CharacterManager.list()]
     return httpdump(CONFIG.get('template', 'index'), entries=entries)