def run(): if len(sys.argv) <= 2: print("At least 2 arguments required (project_file, username).\n") sys.exit() project_file = sys.argv[1] user_name = sys.argv[2] if len(sys.argv) == 4: user_pwd = sys.argv[3] else: user_pwd = '' logger = logging.getLogger('psysmon') logger.setLevel(psysmon.logConfig['level']) logger.addHandler(psysmon.getLoggerHandler()) psybase = create_psybase() userdata = {} userdata['user'] = user_name userdata['pwd'] = user_pwd psybase.load_json_project(project_file, user_name, user_pwd) clear_project_database_tables(psybase.project) psybase.stop_project_server()
def setUpClass(cls): # Configure the logger. logger = logging.getLogger('psysmon') logger.setLevel('DEBUG') logger.addHandler(psysmon.getLoggerHandler(log_level='DEBUG')) cls.data_path = os.path.dirname(os.path.abspath(__file__)) cls.data_path = os.path.join(cls.data_path, 'data') cls.psybase = create_psybase() cls.project = create_empty_project(cls.psybase) cls.project.dbEngine.echo = False clear_project_database_tables(cls.project)
def run(): if len(sys.argv) <= 2: print "At least 2 arguments required (project_file, username).\n" sys.exit() project_file = sys.argv[1] user_name = sys.argv[2] if len(sys.argv) == 4: user_pwd = sys.argv[3] else: user_pwd = '' psybase = create_psybase() userdata = {} userdata['user'] = user_name userdata['pwd'] = user_pwd psybase.load_json_project(project_file, user_name, user_pwd) clear_project_database_tables(psybase.project) psybase.stop_project_server()
def test_write_bulletin_to_database(self): ''' Test the import of a bulletin into the database. ''' bulletin_file = os.path.join(self.data_path, 'bulletin_ims1.0_1.txt') parser = ev_bulletin.ImsParser() parser.parse(bulletin_file) catalog = parser.get_catalog(name='REB', agency_uri='REB') catalog.write_to_database(self.project) db_catalog_orm = self.project.dbTables['event_catalog'] db_session = self.project.getDbSession() result = db_session.query(db_catalog_orm).all() db_session.close() self.assertEqual(len(result), 1) tmp = result[0] self.assertEqual(tmp.name, 'REB') self.assertEqual(len(tmp.events), 1) cur_event = tmp.events[0] self.assertEqual(cur_event.public_id, '112460') self.assertEqual(cur_event.description, 'Southeast of Honshu, Japan') # Clear the database tables. clear_project_database_tables(self.project) bulletin_file = os.path.join(self.data_path, 'bulletin_zamg_ims1.0_1.txt') parser = ev_bulletin.ImsParser() parser.parse(bulletin_file) catalog = parser.get_catalog(name='ZAMG_AUTODRM', agency_uri='ZAMG') catalog.write_to_database(self.project) db_catalog_orm = self.project.dbTables['event_catalog'] db_session = self.project.getDbSession() result = db_session.query(db_catalog_orm).all() db_session.close() self.assertEqual(len(result), 1)
def tearDown(self): clear_project_database_tables(self.project)
def tearDown(self): self.psybase.project_server.unregister_data() clear_project_database_tables(self.project)