Initial commit of final entryHEADmaster

1 files changed, 457 insertions, 0 deletions

diff --git a/lib/gamedb.py b/lib/gamedb.py
new file mode 100644
index 0000000..afbbb48
--- /dev/null
+++ b/lib/gamedb.py
@@ -0,0 +1,457 @@
+# Murder in the Park - A Robotic Mystery
+# Copyright (C) 2008 Jeffrey Armstrong (a.k.a. PrintStar)
+# http://jeff.rainbow-100.com/
+# 
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+# 
+# A full version of the license terms is available in LICENSE.
+
+import sqlite3
+import tempfile
+import shutil
+import os
+from data import *
+
+MOVE_OK = 'Ok'
+MOVE_FAILED = 'Failed'
+
+DEFAULTFILE = 'game.db'
+
+CONTENTS_PREFIX = 'There is a '
+CONTENTS_SUFFIX = ' here.'
+
+def getSafeDirection(string):
+    if string=='n':
+        return 'north'
+    elif string=='s':
+        return 'south'
+    elif string=='e':
+        return 'east'
+    elif string=='w':
+        return 'west'
+    elif string=='u':
+        return 'up'
+    elif string=='d':
+        return 'down'
+    
+
+class GameDB:
+    def __init__(self,filename=None,inplace=False):
+        
+        if not filename:
+            filename = filepath(DEFAULTFILE)
+        
+        self.con = None
+        if not inplace:
+            self.tempname = tempfile.mktemp()
+        
+            shutil.copyfile(filename, self.tempname)
+        
+            self.con = sqlite3.connect(self.tempname)
+        else:
+            self.con = sqlite3.connect(filename)
+            
+        self.con.isolation_level = None
+        
+        self.room = self.getStart()
+    
+    def getRoomName(self):
+        cur = self.con.cursor()
+        cur.execute("select name from Rooms where id=?",(self.room,))
+        for row in cur:
+            return row[0]
+        
+    def getRoomDescription(self):
+        cur = self.con.cursor()
+        cur.execute("select description from Rooms where id=?",(self.room,))
+        for row in cur:
+            return row[0] 
+        
+    def getNorthDescription(self):
+        return self.getDirection("north")
+    
+    def getSouthDescription(self):
+        return self.getDirection("south")
+    
+    def getEastDescription(self):
+        return self.getDirection("east")
+    
+    def getWestDescription(self):
+        return self.getDirection("west")
+    
+    def getUpDescription(self):
+        return self.getDirection("up")
+
+    def getDownDescription(self):
+        return self.getDirection("down")
+    
+    def moveNorth(self):
+        return self.move("north")
+    
+    def moveSouth(self):
+        return self.move("south")
+    
+    def moveEast(self):
+        return self.move("east")
+    
+    def moveWest(self):
+        return self.move("west")
+    
+    def moveUp(self):
+        return self.move("up")
+    
+    def moveDown(self):
+        return self.move("down")
+    
+    def move(self,dir):
+        cur = self.con.cursor()
+        id = self.getDirectionId(dir)
+        if id:
+            
+            if self.getBlock(id):
+                return self.getBlock(id)
+            
+            self.room = id
+            return MOVE_OK
+        return MOVE_FAILED
+    
+    def clearSynonyms(self,string):
+        cur = self.con.cursor()
+        cur.execute('select name from Objects inner join RoomContents on id=objectid where roomid=?',(self.room,))
+        for object in cur:
+            oc = self.con.cursor()
+            oc.execute('select spoken,length(spoken) from Synonyms where name=? order by length(spoken) desc',(object[0],))
+            for item in oc:
+                #print item[0]+':'+object[0]
+                if string.find(' '+item[0]) >= 0:
+                    string = string.replace(' '+item[0],' '+object[0])
+                    break
+                
+        cur.execute('select name from Objects inner join Inventory on Objects.id=Inventory.id')
+        for object in cur:
+            oc = self.con.cursor()
+            oc.execute('select spoken,length(spoken) from Synonyms where name=? order by length(spoken) desc',(object[0],))
+            for item in oc:
+                #print item[0]+':'+object[0]
+                if string.find(' '+item[0]) >= 0:
+                    string = string.replace(' '+item[0],' '+object[0])
+                    break
+                
+        return string
+        
+    def getDirectionId(self,dir):
+        cur = self.con.cursor()
+        cur.execute("select %s from Rooms where id=?" % dir,(self.room,))
+        for row in cur:
+            return row[0]
+        return None
+        
+    def getDirection(self, dir):
+        id = self.getDirectionId(dir)
+        cur = self.con.cursor()
+        if id:        
+            cur.execute("select away_description from Rooms where id=?",(id,))
+            for row in cur:
+                return row[0]
+            
+        return None
+        
+    def getRoomContents(self):
+        rooms = []
+        cur = self.con.cursor()
+        cur.execute("select objectid from RoomContents where roomid=?",(self.room,))
+        for row in cur:
+            oc = self.con.cursor()
+            oc.execute("select name,presence,explicit from Objects where id=?",(row[0],))
+            for row in oc:
+                if row[2]==0:
+                    continue
+                if row[1]:
+                    rooms.append(row[1])
+                else:
+                    rooms.append(CONTENTS_PREFIX+row[0]+CONTENTS_SUFFIX)
+                break
+        return rooms
+    
+    def getInventory(self):
+        inv = []
+        cur = self.con.cursor()
+        cur.execute("select name from Objects outer join Inventory on Objects.id=Inventory.id")
+        for row in cur:
+            inv.append(row[0])
+        return inv
+    
+    def lookAt(self,obj):
+        cur = self.con.cursor()
+        cur.execute("select name,description from RoomContents inner join Objects on Objects.id=RoomContents.objectid where roomid=? and name=?",(self.room,obj))
+        for row in cur:
+            return row[1]
+        return None
+    
+    def pickUp(self,name):
+        cur = self.con.cursor()
+        cur.execute("select id,moveable,person from Objects outer join RoomContents on Objects.id = RoomContents.objectid where name=? and roomid=?",(name,self.room))
+        id = None
+        for obj in cur:
+            id = obj[0]
+            if obj[2] == 1:
+                return "I don't think %s would appreciate that." % name
+            if obj[1] == 0:
+                return "%s can't be moved." % name.title()
+            break
+        
+        if id == None:
+            return None
+        
+        cur.execute("delete from RoomContents where objectid=?",(id,))
+        cur.execute("insert into Inventory(id) values(?)",(id,))
+        return True
+    
+    def drop(self,name):
+        id = None
+        cur = self.con.cursor()
+        cur.execute("select Objects.id from Objects outer join Inventory on Objects.id = Inventory.id where name=?",(name,))
+        for obj in cur:
+            id = obj[0]
+            
+        if id == None:
+            return False
+        
+        cur.execute("delete from Inventory where id=?",(id,))
+        cur.execute("insert into RoomContents(objectid,roomid) values(?,?)",(id,self.room))
+        return True
+    
+    def getObjId(self,name,hereonly=False):
+        cur = self.con.cursor()
+        cur.execute("select Objects.id from Objects outer join Inventory on Objects.id = Inventory.id where name=?",(name,))
+        for obj in cur:
+            return obj[0]
+        cur.execute("select id from Objects outer join RoomContents on Objects.id = RoomContents.objectid where name=? and roomid=?",(name,self.room))
+        for obj in cur:
+            return obj[0]
+        
+        cur.execute("select id from Objects where name=? and everywhere=1",(name,))
+        for obj in cur:
+            return obj[0]
+           
+        if hereonly:
+            return None
+            
+        cur.execute("select id from Objects where name=?",(name,))
+        for obj in cur:
+            return obj[0]
+        
+        return None
+
+    def getBlock(self,id2):
+        cur = self.con.cursor()
+        cur.execute('select state, description from Block where (room1=? and room2=?) or (room1=? and room2=?)',(self.room,id2,id2,self.room))
+        for row in cur:
+            if row[0] == 1:
+                return row[1]
+            else:
+                return None
+        return None
+    
+    def unblock(self,blockid):
+        cur = self.con.cursor()
+        cur.execute('update Block set state=0 where id=?',(blockid,))
+    
+    def getResult(self,obj1,obj2=None,verb=None):
+        id1 = self.getObjId(obj1,True)
+        id2 = None
+        if obj2:
+            id2 = self.getObjId(obj2,True)
+            #print obj2,id2
+        
+        return self.fromCombine(id1,id2,verb,'result')
+
+    def increment(self,incid):
+        cur = self.con.cursor()
+        
+        state = 0
+        max = 1
+        
+        try:
+            incid=int(incid)
+        except ValueError:
+            incid = self.getObjId(incid,False)
+        
+        cur.execute('select state,max from NPC where id=?',(incid,))
+        for row in cur:
+            state = row[0]
+            max = row[1]
+        
+        state = state + 1
+        if state <= max:
+            try:
+                cur.execute('update NPC set state=? where id=?',(state,incid))
+            except:
+                pass
+    
+    def remove(self,incid):
+        try:
+            incid=int(incid)
+        except ValueError:
+            incid = self.getObjId(incid,False)
+            
+        cur = self.con.cursor()
+        cur.execute('delete from Inventory where id=?',(incid,))
+        
+    def to_inv(self,incid):
+        try:
+            incid=int(incid)
+        except ValueError:
+            incid = self.getObjId(incid,False)
+            
+        cur = self.con.cursor()
+        cur.execute('insert into Inventory values(?)',(incid,))
+        
+    def setState(self,incid,state):
+        maxstate = 1
+        try:
+            incid=int(incid)
+        except ValueError:
+            incid = self.getObjId(incid,False)
+        
+        cur = self.con.cursor()  
+        cur.execute('select state,max from NPC where id=?',(incid,))
+        for row in cur:
+            state = max(row[0],state)
+            maxstate = row[1]
+        
+        if state <= maxstate:
+            try:
+                cur.execute('update NPC set state=? where id=?',(state,incid))
+            except:
+                pass
+
+    def getEnding(self):
+        cur = self.con.cursor()  
+        cur.execute("select value from State where name='Ending'")
+        for row in cur:
+            return row[0]
+        return 'End Game'
+        
+    def isEnd(self):
+        cur = self.con.cursor()  
+        cur.execute("select value from State where name='End'")
+        for row in cur:
+            try:
+                id = int(row[0])
+                if self.room == id:
+                    return True
+            except ValueError:
+                return False    
+        return False
+
+    def getIntro(self):
+        cur = self.con.cursor()
+        cur.execute('select value from State where name=?',('Intro',))
+        for row in cur:
+            return row[0]
+        return ' '
+
+    def getTitle(self):
+        cur = self.con.cursor()
+        cur.execute('select value from State where name=?',('Title',))
+        for row in cur:
+            return row[0]
+        return ' '
+
+    def getAction(self,obj1,obj2=None,verb=None):
+        id1 = self.getObjId(obj1,True)
+        id2 = None
+        if obj2:
+            id2 = self.getObjId(obj2,True)
+        
+        return self.fromCombine(id1,id2,verb,'action')
+
+    def getState(self,obj):
+        cur = self.con.cursor()
+        cur.execute('select state from NPC where id=?',(obj,))
+        for row in cur:
+            return row[0]
+        return 0;
+
+    def fromCombine(self,obj1,obj2,verb,column):
+        cur = self.con.cursor()
+        
+        state1 = self.getState(obj1)
+        state2 = self.getState(obj2)
+        state = max(state1,state2)
+        
+        if obj1 and obj2 and verb:
+            cur.execute("select %s from Combine where ((id1=? and id2=?) or (id1=? and id2=?)) and verb=? and state<=? order by state desc" % column,(obj1,obj2,obj2,obj1,verb,state))
+
+        elif obj1 and obj2:
+            cur.execute("select %s from Combine where (id1=? and id2=?) or (id1=? and id2=?) and state<=? order by state desc" % column,(obj1,obj2,obj2,obj1,state))
+
+        elif obj1 and verb:
+            cur.execute("select %s from Combine where id1=? and verb=? and state<=? order by state desc " % column,(obj1,verb,state))
+
+        for row in cur:
+            return row[0]
+        return None
+
+    def checkVerb(self,spoken):
+        if not spoken:
+            return None
+        
+        cur = self.con.cursor()
+        cur.execute("select name from Verbonyms where spoken=?",(spoken,))
+        for row in cur:
+            return row[0]
+        return spoken
+    
+    def use(self,obj1,obj2):
+        id2 = getObjId(obj2,True)
+        if not id1 or not id2:
+            return None     
+        return None
+    
+    def getStart(self):
+        cur = self.con.cursor()
+        cur.execute("select value from State where name='Start'")
+        for row in cur:
+            try:
+                return int(row[0])
+            except ValueError:
+                pass
+            
+        return None
+    
+    def setStart(self):
+        cur = self.con.cursor()
+        cur.execute("update State set value=? where name='Start'",(self.room,))
+        
+    
+    def save(self,filename):
+        self.setStart()
+        self.con.commit()
+        try:
+            shutil.copyfile(self.tempname, filename)
+        except:
+            print "WARN: an error may have occured saving to "+filename
+            return False
+        return True
+        
+    def end(self):
+        self.con.close()
+        
+        try:
+            os.delete(self.tempname)
+        except:
+            pass
+