help='The user id column of check-ins.',
type=str)
arg_parser.add_argument('--venuecol',
default='venue_id',
help='The venue id column of check-ins.',
type=str)
arg_parser.add_argument(
'--userfilter',
help=
'A filter on the selected users (in the format of a SQL WHERE expression).',
type=str)
#arg_parser.add_argument('--plot', action='store_true', help='Pass.')
args = arg_parser.parse_args()
db.config(name=args.dbname,
host=args.dbhost,
user=args.dbuser,
passwd=args.dbpass)
def close_db_terminate():
global db
db.rollback()
db.close()
exit()
def print_table_stats(table, tidcol, classcol):
global db, logger
checkin_count = db.query("SELECT COUNT(*) FROM " + table)[0][0]
traj_count = db.query("SELECT COUNT(DISTINCT(" + tidcol + ")) FROM " +
type=str)
arg_parser.add_argument('config',
help='Configuration file.',
type=str)
args = arg_parser.parse_args()
CREATE_SCHEMA_FILE = 'sql/create_schema.sql'
DROP_SCHEMA_FILE = 'sql/drop_schema.sql'
CONFIG_FILE = args.config
config = configparser.ConfigParser()
config.read(CONFIG_FILE)
db = Database()
db.config(config['DATABASE']['NAME'],
config['DATABASE']['HOST'],
config['DATABASE']['USER'],
config['DATABASE']['PASS'])
if(db.connect()):
logger.log(Logger.INFO, "Succesfully connected to database \'" +
str(config['DATABASE']['NAME']) + "\'!")
else:
logger.log(Logger.ERROR, "Failed connecting to database \'" +
str(config['DATABASE']['NAME']) + "\'!")
if args.operation == 'create':
logger.log(Logger.INFO, "Creating schema for database '" +
str(config['DATABASE']['NAME']) + "'... ")
db.execute(open(CREATE_SCHEMA_FILE, "r").read())
logger.log(Logger.INFO, "Creating schema for database '" +
str(config['DATABASE']['NAME']) + "'... SUCCESS!")
