diff options
Diffstat (limited to 'bin/track_status.py')
-rw-r--r-- | bin/track_status.py | 87 |
1 files changed, 0 insertions, 87 deletions
diff --git a/bin/track_status.py b/bin/track_status.py deleted file mode 100644 index 2a974a5..0000000 --- a/bin/track_status.py +++ /dev/null @@ -1,87 +0,0 @@ -# Copyright (c) 2015 Egor Tensin <Egor.Tensin@gmail.com> -# This file is part of the "VK scripts" project. -# For details, see https://github.com/egor-tensin/vk-scripts. -# Distributed under the MIT License. - -import argparse -import sys - -from vk.api import API -from vk.tracking import StatusTracker -from vk.tracking.db import Format as DatabaseFormat - - -DEFAULT_TIMEOUT = StatusTracker.DEFAULT_TIMEOUT -DEFAULT_DB_FORMAT = DatabaseFormat.CSV - - -def _parse_positive_integer(s): - try: - n = int(s) - except ValueError: - raise argparse.ArgumentTypeError('must be a positive integer: ' + s) - if n < 1: - raise argparse.ArgumentTypeError('must be a positive integer: ' + s) - return n - - -def _parse_database_format(s): - try: - return DatabaseFormat(s) - except ValueError: - raise argparse.ArgumentTypeError('invalid database format: ' + s) - - -def _parse_args(args=None): - if args is None: - args = sys.argv[1:] - - parser = argparse.ArgumentParser( - description='Track when people go online/offline.') - - parser.add_argument('uids', metavar='UID', nargs='+', - help='user IDs or "screen names"') - parser.add_argument('-t', '--timeout', metavar='SECONDS', - type=_parse_positive_integer, - default=DEFAULT_TIMEOUT, - help='set refresh interval') - parser.add_argument('-l', '--log', metavar='PATH', dest='log_path', - help='set log file path (standard output by default)') - parser.add_argument('-f', '--format', dest='db_fmt', - type=_parse_database_format, - choices=DatabaseFormat, - default=DEFAULT_DB_FORMAT, - help='specify database format') - parser.add_argument('-o', '--output', metavar='PATH', dest='db_path', - help='set database file path') - - return parser.parse_args(args) - - -def track_status( - uids, timeout=DEFAULT_TIMEOUT, - log_path=None, - db_path=None, db_fmt=DEFAULT_DB_FORMAT): - - api = API() - tracker = StatusTracker(api, timeout) - - if db_fmt is DatabaseFormat.LOG or db_path is None: - db_fmt = DatabaseFormat.NULL - - 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.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): - track_status(**vars(_parse_args(args))) - - -if __name__ == '__main__': - main() |