def main(): args = arg_parser() st = Setting(args.setting[0]) if args.copy: copy_config(st.default_strings) st.configure() sl = SelectLog(st.setting["put_log_dir"], st.setting["graph_save_dir"], st.setting["log_date_format"]) log_file_paths = sl.get_logfile_paths(args) pg = DrawGraph() for log_file_path in log_file_paths: data, memo = setup_data_frame(args, st.setting, log_file_path) if not data and not memo: continue logfile_name = sl.get_fn(log_file_path) save_dir = sl.setup_save_dir(logfile_name) for stg in st.graph: graph_path = save_dir + "/" + logfile_name + "_" + stg["name"] \ + memo + "." + st.setting["graph_extension"] pg.draw(log_file_path, graph_path, st.setting, stg, data.x_axis, data.df) print("Complete", logfile_name) data.dispose() st.dispose()
def test_get_paths_of_all(self): self.sl = SelectLog("log/", "graph/", "YYMMDDhhmmss") ce.create_exlog(log_date_type=0) all_paths = [f[1] + f[0][2:] + ".csv" for f in ce.default_files] get_all_paths = self.sl.get_paths_of_all() for all_path in all_paths: self.assertIn(all_path, get_all_paths) for get_all_path in get_all_paths: self.assertIn(get_all_path, all_paths)
def test_get_paths_of_all_other_type(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") ce.create_exlog(log_date_type=4) all_paths = [ce.type_four(f) for f in ce.default_files] get_all_paths = self.sl.get_paths_of_all() for all_path in all_paths: self.assertIn(all_path, get_all_paths) for get_all_path in get_all_paths: self.assertIn(get_all_path, all_paths)
def test_get_paths_of_after(self): self.sl = SelectLog("log/", "graph/", "YYMMDDhhmmss") ce.create_exlog(log_date_type=0) after_date = "170102200000" get_after_paths = self.sl.get_paths_of_after(after_date) after_paths = [ f[1] + f[0][2:] + ".csv" for f in ce.default_files if int(f[0][2:]) >= int(after_date) ] for after_path in after_paths: self.assertIn(after_path, get_after_paths) for get_after_path in get_after_paths: self.assertIn(get_after_path, after_paths)
def test_get_paths_of_after_other_type(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") ce.create_exlog(log_date_type=4) after_date = "17-01-02_20-00-00" get_after_paths = self.sl.get_paths_of_after(after_date) after_paths = [] for f in ce.default_files: if int(self.sl.fn_to_datetime(f[0][2:])) >= int( self.sl.fn_to_datetime(after_date)): after_paths.append(ce.type_four(f)) for after_path in after_paths: self.assertIn(after_path, get_after_paths) for get_after_path in get_after_paths: self.assertIn(get_after_path, after_paths)
def test_get_paths_of_select(self): self.sl = SelectLog("log/", "graph/", "YYMMDDhhmmss") ce.create_exlog(log_date_type=0) sel_dates = ["170102200000", "170102000000"] get_sel_paths = self.sl.get_paths_of_select(sel_dates) sel_paths = [] for sel_date in sel_dates: for f in ce.default_files: if int(f[0][2:]) == int(sel_date): sel_paths.append(f[1] + f[0][2:] + ".csv") for sel_path in sel_paths: self.assertIn(sel_path, get_sel_paths) for get_sel_path in get_sel_paths: self.assertIn(get_sel_path, sel_paths)
def test_get_paths_of_select_other_type(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") ce.create_exlog(log_date_type=4) sel_dates = ["17-01-02_20-00-00", "17-01-02_00-00-00"] get_sel_paths = self.sl.get_paths_of_select(sel_dates) sel_paths = [] for sel_date in sel_dates: for f in ce.default_files: if int(self.sl.fn_to_datetime(f[0][2:])) == int( self.sl.fn_to_datetime(sel_date)): sel_paths.append(ce.type_four(f)) for sel_path in sel_paths: self.assertIn(sel_path, get_sel_paths) for get_sel_path in get_sel_paths: self.assertIn(get_sel_path, sel_paths)
def test_get_paths_of_new(self): self.sl = SelectLog("log/", "graph/", "YYMMDDhhmmss") ce.create_exlog(log_date_type=0) if ce.isdir("./graph"): ce.rmtree("./graph") sel_dates = ["170102200000", "170101120000", "170102180000"] get_sel_paths = self.sl.get_paths_of_select(sel_dates) for path in get_sel_paths: self.sl.setup_save_dir(self.sl.get_fn(path)) get_new_paths = self.sl.get_paths_of_new() new_paths = [f[1] + f[0][2:] + ".csv" for f in ce.default_files] for get_sel_path in get_sel_paths: new_paths.remove(get_sel_path) for new_path in new_paths: self.assertIn(new_path, get_new_paths) for get_new_path in get_new_paths: self.assertIn(get_new_path, new_paths)
def test_get_paths_of_new_other_type(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") ce.create_exlog(log_date_type=4) if ce.isdir("./graph"): ce.rmtree("./graph") sel_dates = [ "17-01-02_20-00-00", "17-01-01_12-00-00", "17-01-02_18-00-00" ] get_sel_paths = self.sl.get_paths_of_select(sel_dates) for path in get_sel_paths: self.sl.setup_save_dir(self.sl.get_fn(path)) get_new_paths = self.sl.get_paths_of_new() new_paths = [ce.type_four(f) for f in ce.default_files] for get_sel_path in get_sel_paths: new_paths.remove(get_sel_path) for new_path in new_paths: self.assertIn(new_path, get_new_paths) for get_new_path in get_new_paths: self.assertIn(get_new_path, new_paths)
def test_is_date_in_fn(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") self.assertTrue(self.sl.is_date_in_fn("17-10-31_01-00-00"))
def test_filename_to_date(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") log_file_name = "17-01-02 20,00,00.csv" log_file_date = self.sl.fn_to_datetime(log_file_name) self.assertEqual("170102200000", log_file_date)
class TestSelectLog(unittest.TestCase): def setUp(self): pass def tearDown(self): pass def test_get_paths_of_all(self): self.sl = SelectLog("log/", "graph/", "YYMMDDhhmmss") ce.create_exlog(log_date_type=0) all_paths = [f[1] + f[0][2:] + ".csv" for f in ce.default_files] get_all_paths = self.sl.get_paths_of_all() for all_path in all_paths: self.assertIn(all_path, get_all_paths) for get_all_path in get_all_paths: self.assertIn(get_all_path, all_paths) def test_get_paths_of_after(self): self.sl = SelectLog("log/", "graph/", "YYMMDDhhmmss") ce.create_exlog(log_date_type=0) after_date = "170102200000" get_after_paths = self.sl.get_paths_of_after(after_date) after_paths = [ f[1] + f[0][2:] + ".csv" for f in ce.default_files if int(f[0][2:]) >= int(after_date) ] for after_path in after_paths: self.assertIn(after_path, get_after_paths) for get_after_path in get_after_paths: self.assertIn(get_after_path, after_paths) def test_get_paths_of_select(self): self.sl = SelectLog("log/", "graph/", "YYMMDDhhmmss") ce.create_exlog(log_date_type=0) sel_dates = ["170102200000", "170102000000"] get_sel_paths = self.sl.get_paths_of_select(sel_dates) sel_paths = [] for sel_date in sel_dates: for f in ce.default_files: if int(f[0][2:]) == int(sel_date): sel_paths.append(f[1] + f[0][2:] + ".csv") for sel_path in sel_paths: self.assertIn(sel_path, get_sel_paths) for get_sel_path in get_sel_paths: self.assertIn(get_sel_path, sel_paths) def test_get_paths_of_new(self): self.sl = SelectLog("log/", "graph/", "YYMMDDhhmmss") ce.create_exlog(log_date_type=0) if ce.isdir("./graph"): ce.rmtree("./graph") sel_dates = ["170102200000", "170101120000", "170102180000"] get_sel_paths = self.sl.get_paths_of_select(sel_dates) for path in get_sel_paths: self.sl.setup_save_dir(self.sl.get_fn(path)) get_new_paths = self.sl.get_paths_of_new() new_paths = [f[1] + f[0][2:] + ".csv" for f in ce.default_files] for get_sel_path in get_sel_paths: new_paths.remove(get_sel_path) for new_path in new_paths: self.assertIn(new_path, get_new_paths) for get_new_path in get_new_paths: self.assertIn(get_new_path, new_paths) def test_filename_to_date(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") log_file_name = "17-01-02 20,00,00.csv" log_file_date = self.sl.fn_to_datetime(log_file_name) self.assertEqual("170102200000", log_file_date) def test_get_paths_of_all_other_type(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") ce.create_exlog(log_date_type=4) all_paths = [ce.type_four(f) for f in ce.default_files] get_all_paths = self.sl.get_paths_of_all() for all_path in all_paths: self.assertIn(all_path, get_all_paths) for get_all_path in get_all_paths: self.assertIn(get_all_path, all_paths) def test_is_date_in_fn(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") self.assertTrue(self.sl.is_date_in_fn("17-10-31_01-00-00")) def test_get_paths_of_after_other_type(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") ce.create_exlog(log_date_type=4) after_date = "17-01-02_20-00-00" get_after_paths = self.sl.get_paths_of_after(after_date) after_paths = [] for f in ce.default_files: if int(self.sl.fn_to_datetime(f[0][2:])) >= int( self.sl.fn_to_datetime(after_date)): after_paths.append(ce.type_four(f)) for after_path in after_paths: self.assertIn(after_path, get_after_paths) for get_after_path in get_after_paths: self.assertIn(get_after_path, after_paths) def test_get_paths_of_select_other_type(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") ce.create_exlog(log_date_type=4) sel_dates = ["17-01-02_20-00-00", "17-01-02_00-00-00"] get_sel_paths = self.sl.get_paths_of_select(sel_dates) sel_paths = [] for sel_date in sel_dates: for f in ce.default_files: if int(self.sl.fn_to_datetime(f[0][2:])) == int( self.sl.fn_to_datetime(sel_date)): sel_paths.append(ce.type_four(f)) for sel_path in sel_paths: self.assertIn(sel_path, get_sel_paths) for get_sel_path in get_sel_paths: self.assertIn(get_sel_path, sel_paths) def test_get_paths_of_new_other_type(self): self.sl = SelectLog("log/", "graph/", "YY-MM-DD_hh-mm-ss") ce.create_exlog(log_date_type=4) if ce.isdir("./graph"): ce.rmtree("./graph") sel_dates = [ "17-01-02_20-00-00", "17-01-01_12-00-00", "17-01-02_18-00-00" ] get_sel_paths = self.sl.get_paths_of_select(sel_dates) for path in get_sel_paths: self.sl.setup_save_dir(self.sl.get_fn(path)) get_new_paths = self.sl.get_paths_of_new() new_paths = [ce.type_four(f) for f in ce.default_files] for get_sel_path in get_sel_paths: new_paths.remove(get_sel_path) for new_path in new_paths: self.assertIn(new_path, get_new_paths) for get_new_path in get_new_paths: self.assertIn(get_new_path, new_paths)