vD6hbULr

1def __db_connect(schemacheck=True):
2
3  # connect
4  args = []
5  if dbhost: args.append( ('DBHostName', dbhost) )
6  if dbname: args.append( ('DBName', dbname) )
7  if dbuser: args.append( ('DBUserName', dbuser) )
8  if dbpass: args.append( ('DBPassword', dbpass) )
9  db = MythDB(args=args)
10  log = MythLog(module='RerecordLater', lstr='important')
11
12  # schema version check
13  if schemacheck:
14
15    print 'DB SETTINGS: ' + repr(db.settings.NULL.RerecordLaterDBSchemaVer)
16
17    if db.settings.NULL.RerecordLaterDBSchemaVer == None:
18      # first time run, auto install
19      log.log(
20        MythLog.IMPORTANT,
21        'RerecordLater database does not exist, creating it.'
22      )
23      action_install(db)
24
25------------ snip ---------------
26
27# --install handler
28def action_install(db=None):
29  if not db: db = __db_connect(schemacheck=False)
30  db.cursor().execute("""
31    CREATE TABLE IF NOT EXISTS rerecordlater(
32      chanid INT(10) UNSIGNED,
33      starttime datetime,
34      timeout datetime
35    )
36  """)
37  db.settings.NULL.RerecordLaterDBSchemaVer = schema_version