setup.py

'''
Purpose: create a new database with the name "data.db"
         for storing data and a new table with the db-structure.
Note:    for editing data you can use SQLite data browser
Created on 20.06.2011
@author: Kamil Wozniak
'''

import argparse
import Controller
import sys
import ConfigParser
import DBConnection

if __name__ == '__main__':
    
    if len(sys.argv) < 2:
        print 'Use "-h" or "--help" to view the options.'
        
    parser = argparse.ArgumentParser(description='Setup configures the database connection.')
    parser.add_argument('--rmDev', action='store_true', help='removes all devices from the database')
    parser.add_argument('--rmData', action='store_true', help='removes all saved data from the database')
    parser.add_argument('--rmCMD', action='store_true', help='removes all saved CMD`s from the database')
    parser.add_argument('--sqlite3', action='store_true', help='creates a sqlite3 database called data.db in the same folder')
    parser.add_argument('--mysql', action='store_true', help='creates the mysql tables for a user')
    parser.add_argument('--host', help='mysql: stores the host for the connection')
    parser.add_argument('--user', help='mysql: stores the user for the connection')
    parser.add_argument('--password', help='mysql: stores the password for the connection')
    parser.add_argument('--dbname', help='mysql: stores the databasename for the connection')
    parser.add_help
    
    args = parser.parse_args()
    ''' get the variables from namespace object '''
    argsArray = vars(args)  
    
    if argsArray['rmDev']:
        controller = Controller.Controller()
        controller.removeAllDevicesAction()
        print 'All devices removed.'
        
    if argsArray['rmData']:
        controller = Controller.Controller()
        controller.removeAllDataAction()
        print 'All data removed.'
        
    if argsArray['rmCMD']:
        controller = Controller.Controller()
        controller.removeAllCMDAction()
        print 'All CMD`s removed.'
    
    if argsArray['sqlite3']: 
        # creating/writing the database settings to the configuration file
        configurator = ConfigParser.RawConfigParser()
        configurator.add_section('Database-Config')
        configurator.set('Database-Config', 'type', 'sqlite3')        
        
        with open('wsn.cfg', 'wb') as configfile:
                configurator.write(configfile)
                
        # get the connection object for creating the tables
        connection = DBConnection.DBConnection().getDBConnection()
        cursor = connection.cursor()
            
        cursor.execute("""CREATE TABLE data(
            id TEXT, value TEXT, read INTEGER, createdOn TEXT)""")
        
        cursor.execute("""CREATE TABLE devices(id TEXT, panid TEXT, channel TEXT)""")
        
        cursor.execute("""CREATE TABLE commands(id TEXT, cmd TEXT, 
                          read INTEGER, createdOn TEXT)""")
            
        connection.commit()
        connection.close()
                
        print("SQLite database has been created.")
        
    if argsArray['mysql']:  
        if len(sys.argv) < 10:
            print 'Use "-h" or "--help" to view the options.'
        else:
            # creating/writing the database settings to the configuration file
            configurator = ConfigParser.RawConfigParser()
            configurator.add_section('Database-Config')
            configurator.set('Database-Config', 'type', 'mysql')
            configurator.set('Database-Config', 'host', argsArray['host'])   
            configurator.set('Database-Config', 'user', argsArray['user'])
            configurator.set('Database-Config', 'password', argsArray['password'])  
            configurator.set('Database-Config', 'dbname', argsArray['dbname'])               
            
            with open('wsn.cfg', 'wb') as configfile:
                    configurator.write(configfile)
                    
            # get the connection object for creating the tables
            connection = DBConnection.DBConnection().getDBConnection()
            cursor = connection.cursor()
            
            cursor.execute("""
                CREATE TABLE data(`id` TEXT, `value` TEXT, `read` INTEGER, `createdOn` TEXT)
                """)
            
            cursor.execute("""CREATE TABLE devices(`id` TEXT, `panid` TEXT, `channel` TEXT)""")
            
            cursor.execute("""
                CREATE TABLE commands (`id` TEXT, `cmd` TEXT, `read` INTEGER, `createdOn` TEXT)
                """)
            
                
            connection.commit()
            connection.close()
                    
            print("MySQL tables have been created.")