webgo/goban.py

190 lines
6.2 KiB
Python
Executable file

#!/usr/bin/python
DEBUG = 1
import sys,string
import cgi
import psql,helper,gnugo
picklefile = "goban.pickledump"
###############################################################################
def display_goban(goban,req,form):
"""
gets: dictionary containing the layout of the used goban.
returns: string containing the HTML code for a clickable goban.
"""
data = ""
hoshis19x19 = [(4,4),(4,10),(4,16),(10,4),(10,10),(10,16),(16,4),(16,10),(16,16)]
hoshis13x13 = [(4,4),(4,10),(7,7),(10,4),(10,10)]
hoshis9x9 = [(3,3),(3,7),(5,5),(7,3),(7,7)]
data += '\n<p>\n<form method="post">\n'
data += """<input type="hidden" name="sessionid" value="%s">
<input type="hidden" name="username" value="%s">
<input type="hidden" name="game" value="%s">
""" % (form["sessionid"],form["username"],form["game"])
sgf = psql.get_sgf(form["game"])
helper.debug(req,form,sgf)
gobandict = gnugo.parse_static_gnugo_sgf(sgf)
try:
size = goban["size"]
except:
#TODO: proper error management
raise "[EE] display_goban: got broken goban dictionary."
sys.exit(1)
for x in range(1,size+1):
for y in range(1,size+1):
# check for white or black stone
if gobandict[(x,y)] == 1:
stone = "_white"
elif gobandict[(x,y)] == 2:
stone = "_black"
else:
stone = ""
sx = str(x)
sy = str(y)
# check position:
if (x == 1) and (y == 1): # upper left
data += '<input type=image class="goban" src="img/topleftline'+stone+'.png" name="('+sx+','+sy+')"\n>'
elif (x == 1) and (y == size): # upper right
data += '<input type=image class="goban" src="img/toprightline'+stone+'.png" name="('+sx+','+sy+')"\n><br\n>'
elif (x == size) and (y == size): # lower right
data += '<input type=image class="goban" src="img/bottomrightline'+stone+'.png" name="('+sx+','+sy+')"\n><br>\n'
elif (x == size) and (y == 1): # lower left
data += '<input type=image class="goban" src="img/bottomleftline'+stone+'.png" name="('+sx+','+sy+')"\n>'
elif (y == 1): #left line
data += '<input type=image class="goban" src="img/leftline'+stone+'.png" name="('+sx+','+sy+')"\n>'
elif (x == 1): # top line
data += '<input type=image class="goban" src="img/topline'+stone+'.png" name="('+sx+','+sy+')"\n>'
elif (y == size): # right line
data += '<input type=image class="goban" src="img/rightline'+stone+'.png" name="('+sx+','+sy+')"\n><br\n>'
elif (x == size): #bottom line
data += '<input type=image class="goban" src="img/bottomline'+stone+'.png" name="('+sx+','+sy+')"\n>'
else: # hoshi or empty inner field
defaultfield = '<input type=image class="goban" src="img/centerline'+stone+'.png" name="('+sx+','+sy+')"\n>'
#too lazy to make special images for hoshi fields with stones:
if gobandict[(x,y)] == 1:
hoshifield = '<input type=image class="goban" src="img/centerline_white.png" name="('+sx+','+sy+')"\n>'
elif gobandict[(x,y)] == 2:
hoshifield = '<input type=image class="goban" src="img/centerline_black.png" name="('+sx+','+sy+')"\n>'
else: #empty hoshi
hoshifield = '<input type=image class="goban" src="img/hoshi.png" name="('+sx+','+sy+')"\n>'
if size == 19: # 9 hoshis
if (x,y) in hoshis19x19:
data += hoshifield
else:
data += defaultfield
elif size == 13:
if (x,y) in hoshis13x13:
data += hoshifield
else:
data += defaultfield
elif size == 9:
if (x,y) in hoshis9x9:
data += hoshifield
else:
data += defaultfield
data += '\n<input type="hidden" name="play" value="foo">'
data += '\n</form></p>'
return data
def process_form(req,form,gobandict):
"""
gets a goban dictionary.
reads out the returned CGI form.
if the goban has been clicked, return a (x,y) tuple of the position.
"""
ret = ""
#if form == empty (which means if page is displayed for the first time):
if form.keys() != []:
#cut out the name of the clicked button
for item in form.keys():
if string.find(item,").x")>0:
coordstring = string.split(item,".x")[0]
position = helper.string_to_tuple(coordstring)
ret = set_stone(gobandict, position,req,form)
return ret
def set_stone(gobandict, position,req,form):
"""gets a goban dictionary and a (x,y)-tuple. Returns a modified goban."""
size = gobandict["size"]
name = gobandict["name"]
turn = gobandict["turn_number"]
if (gobandict[position] == 0): #empty field
if gnugo.is_legal(gobandict,position,req,form): #gnugo says the move is ok
#let gnugo make the above move, let gnugo write move to file
new_sgf = gnugo.make_move_in_sgf(req,form,gobandict,position)
#write new sgf file into database
psql.update_goban_table_field(name,"x1","sgf",new_sgf)
psql.update_turn_number(name,turn+1)
psql.set_time(name)
return ""
else: #move not ok
return "This is not a legal move (says Gnugo)."
else: #position not empty
return "Could not make move: Field not empty."
def update_goban_dict_and_table(gobandict,gobanlite):
"""
gets a gobandict and a gobdict light (just coordinates and size).
updates the fields in gobandict and database.
returns changed gobandict and True (or False) if something has been changed (or not).
"""
tf = False
for key in gobanlite.keys():
if gobandict[key] != gobanlite[key]:
#found difference in dicts.
#update gobandict
gobandict[key] = gobanlite[key]
#update databse table. the only valid difference can be changed goban field positions.
psql.update_goban_field(gobandict["name"],key[0],key[1],gobandict[key])
tf = True
return gobandict,tf
###############################################################################
def main(gamename):
# Print the required header that tells the browser how to render the text.
#(currently done by error logger)
#print "Content-Type: text/plain\n\n"
#do stuff
data = helper.header()
#read goban table from database
tmplist = psql.read_table(gamename)
#make a dictionary out of the list
goban = psql.fetchall_list_to_goban_dict(tmplist)
#print goban
(goban,retstring) = process_form(goban)
data += "<p> Turn number: %s, %s Player's Move.</p>" % (goban["turn_number"],("White","Black")[goban["turn_number"] % 2]) #eleet ;>
data += display_goban(goban)
if retstring != "":
data +="\n<p>"+retstring+"</p>\n"
data += helper.footer()
print data