dbHandler.py

#!/usr/bin/python
# -*- coding: utf8 -*-

import sys
import sqlite3
import time
import hashlib
import re

def encodePassword(password):
	return hashlib.md5(password).hexdigest()

def isValidName(name):
	if len(name)==0:
		return False
	if len(name)>20:
		return False
	if re.match("^[A-Za-z0-9_-]*$", name):
		return True
	return False

def isValidPassword(password):
	if len(password)<4:
		return False
	return True

class DbHandler(object):
	def __init__(self, dbfile):
		self.dbfile = dbfile
		self.connection = sqlite3.connect(self.dbfile)
		self.cursor = self.connection.cursor()
		
	def execute(self, args):
		#print "args were : %s" % (args)
		self.cursor.execute(args)
		result = self.cursor.fetchall()
		self.connection.commit()
		return result
	
	#-------------------------------------------------------------------
	# game db init
	#-------------------------------------------------------------------	
	def initTables(self):
		self.createPlayersTable()
		self.createCharTable()
		self.createItemTable()
		self.createInventoryTable()
		self.createStorageTable()
		
	def reset(self):
		self.execute("""drop table players""")
		self.execute("""drop table characters""")
		self.execute("""drop table items""")
		self.execute("""drop table inventory""")
		self.execute("""drop table storage""")
	
	#-------------------------------------------------------------------
	# players table
	#-------------------------------------------------------------------
		
	def createPlayerTable(self):
		self.execute("""create table players(
			id integer primary key autoincrement,
			name text,
			password text,
			date text)""")
		
	def addPlayer(self, name, password):
		if self.hasPlayer(name):
			print "Couldn't create player %s : that name already exists." % (name)
			return False
		if len(name)>30:
			print "Couldn't create player %s : that name is way too long!" % (name)
			return False
			
		if " " in name:
			print "Couldn't create player %s : no space allowed in name!" % (name)
			return False
		
		today = time.strftime("%Y%m%d%H%M")
		msg = """insert into players values(NULL, '%s', '%s', '%s')""" % (name, encodePassword(password), today)
		self.execute(msg)
		print "Created player %s" % (name)
		
		
	def delPlayer(self, name):
		msg = """delete from players where name='%s'""" % (name)
		self.execute(msg)
		
	def getPlayer(self, name):
		info = self.execute("""select * from players where name = '%s'""" % (name))
		if len(info)>0:
			return info[0]
		return None
		
	
	def hasPlayer(self, name):
		if len(self.execute("""select * from players where name = '%s'""" % (name)))>0:
			return True
		return False
	
	def checkLogin(self, name, password):
		res = self.execute("""select * from players where name = '%s'""" % (name))
		if len(res)>0:
			if res[0][2]==encodePassword(password):
				return True
		return False
	
	#-------------------------------------------------------------------
	# characters
	#-------------------------------------------------------------------
	
	def createCharTable(self):
		self.execute("""create table characters(
			playerName integer,
			name text,
			sex text,
			gmLevel integer,
			map text,
			x text,
			y text
			)""")
	
	#-------------------------------------------------------------------
	# items, equipment, inventory, storage
	#-------------------------------------------------------------------
	
	def createItemTable(self):
		self.execute("""create table items(
				itemId integer,
				name text,
				category text,
				weight text
				
			)""")
		
	def createInventoryTable(self):
		self.execute("""create table inventory(
			ownerName text,
			itemId integer,
			quantity integer
			)""")
			
	def createStorageTable(self):
		self.execute("""create table storage(
			ownerName text,
			itemId integer,
			quantity integer
			)""")
			
#dbhandler = DbHandler("db/essai.db")