aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/vk/tracking/track_status.py
diff options
context:
space:
mode:
authorEgor Tensin <Egor.Tensin@gmail.com>2021-05-03 21:55:12 +0300
committerEgor Tensin <Egor.Tensin@gmail.com>2021-05-03 21:55:12 +0300
commitb3c502ffd0a7feb158e9efdbfcf9bd62ce0393b4 (patch)
treebc29df6499e44bd11957f02a763f48f65c593a48 /vk/tracking/track_status.py
parentvk.tracking: status_tracker -> status (diff)
downloadvk-scripts-b3c502ffd0a7feb158e9efdbfcf9bd62ce0393b4.tar.gz
vk-scripts-b3c502ffd0a7feb158e9efdbfcf9bd62ce0393b4.zip
vk.tracking: merge *_status to status.py
Diffstat (limited to 'vk/tracking/track_status.py')
-rw-r--r--vk/tracking/track_status.py87
1 files changed, 0 insertions, 87 deletions
diff --git a/vk/tracking/track_status.py b/vk/tracking/track_status.py
deleted file mode 100644
index 2a974a5..0000000
--- a/vk/tracking/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()