added main method and a readme

Author: davidruffner

Diff for: seatingchart/README.md

@@ -0,0 +1,6 @@
+Seating Chart readme
+
+This program creates a seating chart from a list of guests and the friendships between the guests.
+
+Usage from the terminal:
+$ python -m seatingchart.seatingchart <excelfilein> <excelfileout>

Diff for: seatingchart/example/testseatingchart3.py

@@ -19,7 +19,7 @@
 org = sc.Organizer(guestlist, chart)
 org0 = sc.Organizer(guestlist, chart0)
 # Seat everyone
-org.seatguests()
+org.seatguestsfriends()
 
 print "The final seating chart"
 chart.print_seatingchart()

Diff for: seatingchart/seatingchart.py

@@ -1,6 +1,7 @@
 import random
 import numpy as np
 import pandas
+import matplotlib.pyplot as plt
 
 """
 This simple program will try to sort out a seating chart from a list of
@@ -139,6 +140,9 @@ def pick_friend_of(self, guestname):
 
     def seat_guest(self, guestname):
         n = guestname
+        if len(self.open) == 0:
+            print "No more seats!"
+            return
         seatindex = self.open.pop(random.randrange(len(self.open)))
         self.chart[seatindex[0]][seatindex[1]] = n
 
@@ -148,19 +152,18 @@ def seat_guest(self, guestname):
         self.toseat.remove(n)
 
     def seat_guest_here(self, guestname, index):
-        n = guestname
         # check if the seat is open
         if self.open.count(index) == 0:
             print "the seat is already filled!!"
             return None
         else:
             self.open.remove(index)
-        self.chart[index[0]][index[1]] = n
+        self.chart[index[0]][index[1]] = guestname
 
         # update the seatdict, and seated
-        self.seated.append(n)
-        self.seatdict[n] = index
-        self.toseat.remove(n)
+        self.seated.append(guestname)
+        self.seatdict[guestname] = index
+        self.toseat.remove(guestname)
 
     def seat_guestfriend(self, guestname, friendname):
         """
@@ -355,3 +358,55 @@ def metropolisstep(self, temp=0.1):
             else:
                 print "it's temporarily worse but keep it"
                 return friendcount1
+
+def main(excelfilein,excelfileout):
+    guestlist = GuestList()
+    guestlist.fromExcel(excelfilein)
+    guestlist.print_friendlist()
+
+
+    nguests = len(guestlist.guests)
+    print nguests
+    seatsper = 8
+    print nguests/seatsper
+
+    chart = SeatingChart(guestlist, nguests/seatsper+1, seatsper)
+    org = Organizer(guestlist, chart)
+    # Seat everyone
+    print "Seat guests with a friend"
+    org.seatguestsfriends()
+    print "now do Metropolis algorithim to increase friendships"
+    count0 = org.friendcount()
+    nsteps = 1000
+    friendsT1 = np.zeros(nsteps)
+    steps = np.arange(nsteps)
+    for i in steps:
+        print i
+        friendsT1[i] = org.metropolisstep(temp=0.01)
+    count1 = org.friendcount()
+    print "Metropolis algorithm added {0} friends".format(count1-count0)
+    print "export seating chart to an excel file"
+    plt.plot(steps, friendsT1, label='T1')
+    plt.xlabel = "step"
+    plt.ylabel = "friendships"
+    plt.show()
+    org.sc.to_excel(excelfileout)
+
+
+if __name__ == '__main__':
+    import sys
+    umesg = "python -m seatingchart.seatingchar excelfilein (excelfileout)"
+    if len(sys.argv) > 1:
+        # print 'nothing here'
+        if len(sys.argv) > 2:
+            excelfilein = sys.argv[1]
+            excelfileout = sys.argv[2]
+            main(excelfilein, excelfileout)
+        else:
+            excelfilein = sys.argv[1]
+            excelfileout = excelfilein[:-5]+"out"+excelfilein[-5:]
+            main(excelfilein, excelfileout)
+    else:
+        print umesg
+
+