fix Pylint warnings

3 files changed, 21 insertions, 15 deletions

diff --git a/bin/mutual_friends.py b/bin/mutual_friends.py
index 8e6e584..d8591cf 100644
--- a/bin/mutual_friends.py
+++ b/bin/mutual_friends.py
@@ -76,7 +76,10 @@ def _parse_output_format(s):
     except ValueError:
         raise argparse.ArgumentTypeError('invalid output format: ' + s)
 
-def _parse_args(args=sys.argv):
+def _parse_args(args=None):
+    if args is None:
+        args = sys.argv[1:]
+
     parser = argparse.ArgumentParser(
         description='Learn who your ex and her new boyfriend are both friends with.')
 
@@ -92,7 +95,7 @@ def _parse_args(args=sys.argv):
                         default=sys.stdout,
                         help='set output file path (standard output by default)')
 
-    return parser.parse_args(args[1:])
+    return parser.parse_args(args)
 
 def write_mutual_friends(uids, fmt=OutputFormat.CSV, fd=sys.stdout):
     api = API(Language.EN)
@@ -104,9 +107,8 @@ def write_mutual_friends(uids, fmt=OutputFormat.CSV, fd=sys.stdout):
     with fmt.create_writer(fd) as writer:
         writer.write_mutual_friends(mutual_friends)
 
-def main(args=sys.argv):
-    args = _parse_args(args)
-    write_mutual_friends(**vars(args))
+def main(args=None):
+    write_mutual_friends(**vars(_parse_args(args)))
 
 if __name__ == '__main__':
     main()
diff --git a/bin/online_sessions.py b/bin/online_sessions.py
index ac182e7..153878f 100644
--- a/bin/online_sessions.py
+++ b/bin/online_sessions.py
@@ -379,7 +379,10 @@ def _parse_date_range_limit(s):
         raise argparse.ArgumentTypeError(
             msg.format(_DATE_RANGE_LIMIT_FORMAT, s))
 
-def _parse_args(args=sys.argv):
+def _parse_args(args=None):
+    if args is None:
+        args = sys.argv[1:]
+
     parser = argparse.ArgumentParser(
         description='View/visualize the amount of time people spend online.')
 
@@ -412,7 +415,7 @@ def _parse_args(args=sys.argv):
                         type=_parse_date_range_limit, default=None,
                         help='discard online activity after this moment')
 
-    return parser.parse_args(args[1:])
+    return parser.parse_args(args)
 
 def process_online_sessions(
         db_fd, db_fmt=DatabaseFormat.CSV,
@@ -429,9 +432,8 @@ def process_online_sessions(
         output_writer.process_database(
             group_by, db_reader, time_from=time_from, time_to=time_to)
 
-def main(args=sys.argv):
-    args = _parse_args(args)
-    process_online_sessions(**vars(args))
+def main(args=None):
+    process_online_sessions(**vars(_parse_args(args)))
 
 if __name__ == '__main__':
     main()
diff --git a/bin/track_status.py b/bin/track_status.py
index c221c53..d8d8908 100644
--- a/bin/track_status.py
+++ b/bin/track_status.py
@@ -27,7 +27,10 @@ def _parse_database_format(s):
     except ValueError:
         raise argparse.ArgumentTypeError('invalid database format: ' + s)
 
-def _parse_args(args=sys.argv):
+def _parse_args(args=None):
+    if args is None:
+        args = sys.argv[1:]
+
     parser = argparse.ArgumentParser(
         description='Track when people go online/offline.')
 
@@ -51,7 +54,7 @@ def _parse_args(args=sys.argv):
                         default=None,
                         help='set database file path')
 
-    return parser.parse_args(args[1:])
+    return parser.parse_args(args)
 
 def track_status(
         uids, timeout=DEFAULT_TIMEOUT,
@@ -70,9 +73,8 @@ def track_status(
             tracker.add_database_writer(db_writer)
             tracker.loop(uids)
 
-def main(args=sys.argv):
-    args = _parse_args(args)
-    track_status(**vars(args))
+def main(args=None):
+    track_status(**vars(_parse_args(args)))
 
 if __name__ == '__main__':
     main()