see ChangeLog

Monotone-Revision: 9054022ef1ca8aeba6e34842d27d9b94ce002b89

Monotone-Author: dev-unix.inverse.qc.ca
Monotone-Date: 2006-06-15T19:34:10
Monotone-Branch: ca.inverse.sogo

Misc/db/scripts/create_user_tables.py

@@ -0,0 +1,161 @@
+#!/usr/bin/env python
+
+"""
+README!!!
+
+Check the table names in
+CREATE_QUICK + CREATE_BLOB -> are the table names correct?
+"""
+
+import pg
+import sys
+import random
+
+
+# the default database. this is where folder_info lives
+DEFAULT_DB="test"
+USER="OGo"
+PASSWORD="OGo"
+HOST="agenor-db"
+PORT="5432"
+
+DEFAULT_CONNECTION = "%s:%s@%s:%s" % (USER, PASSWORD, HOST, PORT)
+
+# insert more names if you want to test with multiple db's!
+dbConnectionMap = { DEFAULT_DB : DEFAULT_CONNECTION }
+
+
+# lookup table for db connections
+connectionPool = {}
+
+
+#
+# TEMPLATES
+#
+
+CREATE_QUICK="""CREATE TABLE SOGo_%(user)s_privcal_quick (
+  c_name       VARCHAR(256)    NOT NULL PRIMARY KEY,
+  uid          VARCHAR(256)    NOT NULL,
+  startdate    INT NOT NULL,
+  enddate      INT NOT NULL,
+  title        VARCHAR(1000)   NOT NULL,
+  participants VARCHAR(100000) NOT NULL
+)"""
+
+CREATE_BLOB="""CREATE TABLE SOGo_%(user)s_privcal (
+  c_name         VARCHAR(256)    NOT NULL PRIMARY KEY,
+  c_content      VARCHAR(100000) NOT NULL,
+  c_creationdate INT             NOT NULL,
+  c_lastmodified INT             NOT NULL,
+  c_version      INT             NOT NULL
+)"""
+
+INSERT_FOLDERINFO_USER="""INSERT INTO SOGo_folder_info 
+  ( c_path, c_path1, c_path2, c_path3, c_path4, c_foldername, 
+    c_location, c_quick_location, c_folder_type ) 
+VALUES 
+  ( '/Users/%(user)s', 
+    'Users',
+    '%(user)s',
+    NULL,
+    NULL,
+    '%(user)s', 
+    'http://%(connection)s/%(db)s/SOGo_user_folder', 
+    'http://%(connection)s/%(db)s/SOGo_user_folder_quick', 
+    'Container'
+)"""
+
+INSERT_FOLDERINFO_USER_PRIVCAL="""INSERT INTO SOGo_folder_info 
+  ( c_path, c_path1, c_path2, c_path3, c_path4, c_foldername, 
+    c_location, c_quick_location, c_folder_type ) 
+VALUES 
+  ( '/Users/%(user)s/Calendar', 
+    'Users',
+    '%(user)s',
+    'Calendar',
+     NULL,
+    'Calendar', 
+    'http://%(connection)s/%(db)s/SOGo_%(user)s_privcal', 
+    'http://%(connection)s/%(db)s/SOGo_%(user)s_privcal_quick', 
+    'Appointment'
+)"""
+
+
+#
+# HELPERS
+#
+
+def usage():
+  sys.stderr.write("""create_user_tables.py <filename>
+  
+  filename is plain text, platform encoding. Each line has a username. Simple.
+  """)
+  sys.exit(1)
+
+def splitConnectionString(connectionString):
+  res = []
+  for t in connectionString.split("@"):
+    for r in t.split(":"):
+      res.append(r)
+  return res
+  
+#
+# MAIN
+#
+
+def run():
+  global dbConnectionMap, connectionPool
+
+  # check arguments
+  try:
+    filename = sys.argv[1]
+  except IndexError:
+    usage()
+
+  # check filename
+  try:
+    f = open (filename, "r");
+  except IOError:
+    sys.stderr.write("%s\n" % (sys.exc_value))
+    sys.exit(1)
+
+  # setup connection pools
+  dbs = dbConnectionMap.keys()
+  dbsCount = len(dbs)
+  for db in dbs:
+    conString = dbConnectionMap[db]
+    c = splitConnectionString(conString)
+    #pg.DB(dbname, host, port, opt, tty, user, passwd)
+    con = pg.DB(db, c[2], int(c[3]), None, None, c[0], c[1])
+    connectionPool[db] = con
+
+  # get the default connection
+  db = DEFAULT_DB
+  defcon = connectionPool[db]
+  con = defcon
+
+  # read list of users
+  users = f.readlines()
+  for user in users:
+    user = user.strip()
+    
+    # pick a random database if dbsCount > 1
+    if dbsCount > 1:
+      idx = random.randrange(0, dbsCount)
+      db = dbs[idx]
+      con = connectionPool[db]
+  
+    map = { "user"       : user,
+            "db"         : db,
+            "connection" : dbConnectionMap[db],
+          }
+   
+    con.query(CREATE_QUICK % map)
+    con.query(CREATE_BLOB % map)
+    defcon.query(INSERT_FOLDERINFO_USER % map)
+    defcon.query(INSERT_FOLDERINFO_USER_PRIVCAL % map)
+
+
+# let the games begin ...
+if __name__ == "__main__":
+    run()