diff options
author | Egor Tensin <Egor.Tensin@gmail.com> | 2017-01-28 02:40:27 +0300 |
---|---|---|
committer | Egor Tensin <Egor.Tensin@gmail.com> | 2017-01-28 02:40:27 +0300 |
commit | 1c0e2c13d4f0acbe42e43d886df4f1067506efed (patch) | |
tree | 8d76116d718293275315605b66f1466af7d09c27 /bin | |
parent | bin: move file i/o to a separate module (diff) | |
download | vk-scripts-1c0e2c13d4f0acbe42e43d886df4f1067506efed.tar.gz vk-scripts-1c0e2c13d4f0acbe42e43d886df4f1067506efed.zip |
vk: move file i/o to a separate module
Diffstat (limited to 'bin')
-rw-r--r-- | bin/online_sessions.py | 3 | ||||
-rw-r--r-- | bin/track_status.py | 7 |
2 files changed, 7 insertions, 3 deletions
diff --git a/bin/online_sessions.py b/bin/online_sessions.py index cb3f4bf..be3a4f2 100644 --- a/bin/online_sessions.py +++ b/bin/online_sessions.py @@ -335,7 +335,8 @@ def process_online_sessions( if time_from > time_to: time_from, time_to = time_to, time_from - with db_fmt.create_reader(db_path) as db_reader: + with db_fmt.open_input_file(db_path) as db_fd: + db_reader = db_fmt.create_reader(db_fd) with out_fmt.open_file(out_path) as out_fd: out_sink = out_fmt.create_sink(out_fd) out_sink.process_database( diff --git a/bin/track_status.py b/bin/track_status.py index 404c1ec..4c679c5 100644 --- a/bin/track_status.py +++ b/bin/track_status.py @@ -63,10 +63,13 @@ def track_status( if db_fmt is DatabaseFormat.LOG or db_path is None: db_fmt = DatabaseFormat.NULL - with DatabaseFormat.LOG.create_writer(log_path) as log_writer: + with DatabaseFormat.LOG.open_output_file(log_path) as log_fd: + log_writer = DatabaseFormat.LOG.create_writer(log_fd) tracker.add_database_writer(log_writer) - with db_fmt.create_writer(db_path) as db_writer: + with db_fmt.open_output_file(db_path) as db_fd: + db_writer = db_fmt.create_writer(db_fd) tracker.add_database_writer(db_writer) + tracker.loop(uids) def main(args=None): |