home *** CD-ROM | disk | FTP | other *** search
/ linuxmafia.com 2016 / linuxmafia.com.tar / linuxmafia.com / bale / baleit.py < prev    next >
Text File  |  2005-02-10  |  2KB  |  68 lines
  1. import os
  2. import string
  3. import MySQLdb  
  4.  
  5. def opendb():
  6.  host = 'localhost'
  7.  db  = 'events'
  8.   
  9.  user = 'bale'                                                                
  10.  passwd = 'cabalista'                                                         
  11.  rdb = MySQLdb.connect(db=db, host=host, user=user, passwd=passwd)            
  12.  cursor = rdb.cursor()
  13.  return rdb, cursor
  14.  
  15. def getfilelist():
  16.  pwd = "%s/groups/" % (os.getcwd())
  17.  fl = os.listdir(pwd)
  18.  
  19.  return pwd, fl
  20.  
  21. def parsefile(fname):
  22.  f = open(fname, 'r')
  23.  g = f.readlines()
  24.  f.close()
  25.  return g
  26.  
  27. def fudgeQuotes(str):
  28.  g = str.strip()
  29.  g = string.replace(g, "'", "''")
  30.  return g
  31.  
  32. def getfiles(cursor):
  33.  pwd, fl = getfilelist()
  34.  for i in fl:
  35.   lines = parsefile("%s%s" % (pwd, i))
  36.         
  37.   q1 = "select count(*) from groups where gcode = '%s'" % (i,)
  38.   q3 = "delete from groups where gcode = '%s'" % (i,)
  39.  
  40.   q2 = "insert into groups (gcode, gname, gcity, gcounty, "
  41.   q2 = q2 + "active, outofarea, description) values "
  42.   q2 = q2 + "('%s', '%s', '%s', '%s', %s, %s, '%s')"
  43.  
  44.   cursor.execute(q1)
  45.   gcode   = i
  46.   gname   = fudgeQuotes(lines[0])
  47.   gcity   = fudgeQuotes(lines[1])
  48.   gcounty = fudgeQuotes(lines[2])
  49.   gactive = fudgeQuotes(lines[3])
  50.   goutofarea = fudgeQuotes(lines[4])
  51.   description = fudgeQuotes(string.join(lines[5:], ''))
  52.  
  53.   resultSet = cursor.fetchall()
  54.   if resultSet[0][0] != 0:
  55.    cursor.execute(q3)
  56.   
  57.  
  58.   stmt = q2 % (gcode, gname, gcity, gcounty, gactive, goutofarea, description)
  59.  
  60.   cursor.execute(stmt)
  61.  
  62.  
  63.   print "group %s added successfully" % (i,)          
  64.  
  65. if __name__ == "__main__":
  66.  db, cursor = opendb()
  67.  getfiles(cursor)
  68.