Refactor code to comply with Black style

Author: DeD1rk

scripts/fuzzer.py

@@ -7,15 +7,13 @@
 
 from watson import Watson
 
-if not os.environ.get('WATSON_DIR'):
+if not os.environ.get("WATSON_DIR"):
     sys.exit(
         "This script will corrupt Watson's data, please set the WATSON_DIR "
         "environment variable to safely use it for development purpose."
     )
 
-watson = Watson(config_dir=os.environ.get('WATSON_DIR'),
-                frames=None,
-                current=None)
+watson = Watson(config_dir=os.environ.get("WATSON_DIR"), frames=None, current=None)
 
 projects = [
     ("apollo11", ["reactor", "module", "wheels", "steering", "brakes"]),
@@ -26,21 +24,22 @@
 
 now = arrow.now()
 
-for date in arrow.Arrow.range('day', now.shift(months=-1), now):
+for date in arrow.Arrow.range("day", now.shift(months=-1), now):
     if date.weekday() in (5, 6):
         # Weekend \o/
         continue
 
-    start = date.replace(hour=9, minute=random.randint(0, 59)) \
-                .shift(seconds=random.randint(0, 59))
+    start = date.replace(hour=9, minute=random.randint(0, 59)).shift(
+        seconds=random.randint(0, 59)
+    )
 
     while start.hour < random.randint(16, 19):
         project, tags = random.choice(projects)
         frame = watson.frames.add(
             project,
             start,
             start.shift(seconds=random.randint(60, 4 * 60 * 60)),
-            tags=random.sample(tags, random.randint(0, len(tags)))
+            tags=random.sample(tags, random.randint(0, len(tags))),
         )
         start = frame.stop.shift(seconds=random.randint(0, 1 * 60 * 60))
 

scripts/gen-cli-docs.py

@@ -5,67 +5,65 @@
 from click.core import Command, Context
 from click.formatting import HelpFormatter
 from watson import cli as watson_cli
+
 # from watson import watson
 
 
 class MarkdownFormatter(HelpFormatter):
-
     def write_heading(self, heading):
         """Writes a heading into the buffer."""
-        self.write('### {}\n'.format(heading))
+        self.write("### {}\n".format(heading))
 
-    def write_usage(self, prog, args='', prefix='Usage: '):
+    def write_usage(self, prog, args="", prefix="Usage: "):
         """Writes a usage line into the buffer.
         :param prog: the program name.
         :param args: whitespace separated list of arguments.
         :param prefix: the prefix for the first line.
         """
-        self.write('```bash\n{} {} {}\n```\n'.format(prefix, prog, args))
+        self.write("```bash\n{} {} {}\n```\n".format(prefix, prog, args))
 
     def write_text(self, text):
-        """Writes re-indented text into the buffer.
-        """
+        """Writes re-indented text into the buffer."""
 
         should_indent = False
         rows = []
 
-        for row in text.split('\n'):
+        for row in text.split("\n"):
 
             if should_indent:
-                row = '    {}'.format(row)
+                row = "    {}".format(row)
 
-            if '\b' in row:
-                row = row.replace('\b', '', 1)
+            if "\b" in row:
+                row = row.replace("\b", "", 1)
                 should_indent = True
             elif not len(row.strip()):
                 should_indent = False
 
             rows.append(row)
 
-        self.write("{}\n".format('\n'.join(rows)))
+        self.write("{}\n".format("\n".join(rows)))
 
     def write_dl(self, rows, **kwargs):
         """Writes a definition list into the buffer.  This is how options
         and commands are usually formatted.
         :param rows: a list of two item tuples for the terms and values.
         """
         rows = list(rows)
-        self.write('\n')
+        self.write("\n")
 
-        self.write('Flag | Help\n')
-        self.write('-----|-----\n')
+        self.write("Flag | Help\n")
+        self.write("-----|-----\n")
 
         for row in rows:
-            self.write('`{}` | {}\n'.format(*row))
-        self.write('\n')
+            self.write("`{}` | {}\n".format(*row))
+        self.write("\n")
 
 
 class MkdocsContext(Context):
-
     @property
     def command_path(self):
         # Not so proud of it
-        return 'watson {}'.format(self.command.name)
+        return "watson {}".format(self.command.name)
 
     def make_formatter(self):
         return MarkdownFormatter()
@@ -84,18 +82,19 @@ def is_click_command(obj):
             return True
         return False
 
-    content = '\n'.join((
-        "<!-- ",
-        "    This document has been automatically generated.",
-        "    It should NOT BE EDITED.",
-        "    To update this part of the documentation,",
-        "    please type the following from the repository root:",
-        "    $ make docs"
-        "-->",
-        "",
-        "# Commands",
-        "",
-    ))
+    content = "\n".join(
+        (
+            "<!-- ",
+            "    This document has been automatically generated.",
+            "    It should NOT BE EDITED.",
+            "    To update this part of the documentation,",
+            "    please type the following from the repository root:",
+            "    $ make docs" "-->",
+            "",
+            "# Commands",
+            "",
+        )
+    )
 
     # Iterate over commands to build docs
     for cmd_name, cmd in inspect.getmembers(watson_cli, is_click_command):
@@ -106,14 +105,14 @@ def is_click_command(obj):
 
         # Each command is a section
         content += "## `{}`\n\n".format(cmd_name)
-        content += ''.join(formatter.buffer)
+        content += "".join(formatter.buffer)
 
     # Write the commands documentation file
-    with open(rowsput, 'w') as f:
+    with open(rowsput, "w") as f:
         f.write(content)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
 
-    commands_md = 'docs/user-guide/commands.md'
+    commands_md = "docs/user-guide/commands.md"
     main(commands_md)

setup.py

@@ -5,16 +5,16 @@
 
 from setuptools import setup
 
-with open('README.rst') as f:
+with open("README.rst") as f:
     readme = f.read()
 
 # read package meta-data from version.py
 pkg = {}
-mod = join('watson', 'version.py')
-exec(compile(open(mod).read(), mod, 'exec'), {}, pkg)
+mod = join("watson", "version.py")
+exec(compile(open(mod).read(), mod, "exec"), {}, pkg)
 
 
-def parse_requirements(requirements, ignore=('setuptools',)):
+def parse_requirements(requirements, ignore=("setuptools",)):
     """Read dependencies from requirements file (with version numbers if any)
 
     Note: this implementation does not support requirements files with extra
@@ -24,32 +24,32 @@ def parse_requirements(requirements, ignore=('setuptools',)):
         packages = set()
         for line in f:
             line = line.strip()
-            if line.startswith(('#', '-r', '--')):
+            if line.startswith(("#", "-r", "--")):
                 continue
-            if '#egg=' in line:
-                line = line.split('#egg=')[1]
+            if "#egg=" in line:
+                line = line.split("#egg=")[1]
             pkg = line.strip()
             if pkg not in ignore:
                 packages.add(pkg)
         return tuple(packages)
 
 
 setup(
-    name='td-watson',
-    version=pkg['version'],
-    description='A wonderful CLI to track your time!',
+    name="td-watson",
+    version=pkg["version"],
+    description="A wonderful CLI to track your time!",
     url="https://github.com/TailorDev/Watson",
-    packages=['watson'],
-    author='TailorDev',
-    author_email='contact@tailordev.fr',
-    license='MIT',
+    packages=["watson"],
+    author="TailorDev",
+    author_email="contact@tailordev.fr",
+    license="MIT",
     long_description=readme,
-    install_requires=parse_requirements('requirements.txt'),
-    python_requires='>=3.6',
-    tests_require=parse_requirements('requirements-dev.txt'),
+    install_requires=parse_requirements("requirements.txt"),
+    python_requires=">=3.6",
+    tests_require=parse_requirements("requirements-dev.txt"),
     entry_points={
-        'console_scripts': [
-            'watson = watson.__main__:cli',
+        "console_scripts": [
+            "watson = watson.__main__:cli",
         ]
     },
     classifiers=[
@@ -73,5 +73,5 @@ def parse_requirements(requirements, ignore=('setuptools',)):
         "Topic :: Office/Business",
         "Topic :: Utilities",
     ],
-    keywords='watson time-tracking time tracking monitoring report',
+    keywords="watson time-tracking time tracking monitoring report",
 )

tests/__init__.py

@@ -8,23 +8,18 @@
 import py
 
 
-TEST_FIXTURE_DIR = py.path.local(
-    os.path.dirname(
-        os.path.realpath(__file__)
-        )
-    ) / 'resources'
+TEST_FIXTURE_DIR = (
+    py.path.local(os.path.dirname(os.path.realpath(__file__))) / "resources"
+)
 
 
 def mock_datetime(dt, dt_module):
-
     class DateTimeMeta(type):
-
         @classmethod
         def __instancecheck__(mcs, obj):
             return isinstance(obj, datetime.datetime)
 
     class BaseMockedDateTime(datetime.datetime):
-
         @classmethod
         def now(cls, tz=None):
             return dt.replace(tzinfo=tz)
@@ -37,9 +32,9 @@ def utcnow(cls):
         def today(cls):
             return dt
 
-    MockedDateTime = DateTimeMeta('datetime', (BaseMockedDateTime,), {})
+    MockedDateTime = DateTimeMeta("datetime", (BaseMockedDateTime,), {})
 
-    return mock.patch.object(dt_module, 'datetime', MockedDateTime)
+    return mock.patch.object(dt_module, "datetime", MockedDateTime)
 
 
 def mock_read(content):

tests/conftest.py

@@ -8,7 +8,7 @@
 
 @pytest.fixture
 def config_dir(tmpdir):
-    return str(tmpdir.mkdir('config'))
+    return str(tmpdir.mkdir("config"))
 
 
 @pytest.fixture

tests/test_autocompletion.py

@@ -42,9 +42,7 @@
         (get_tags, None, []),
     ],
 )
-def test_if_returned_values_are_distinct(
-    watson_df, func_to_test, rename_type, args
-):
+def test_if_returned_values_are_distinct(watson_df, func_to_test, rename_type, args):
     ctx = ClickContext(obj=watson_df, params={"rename_type": rename_type})
     prefix = ""
     ret_list = list(func_to_test(ctx, args, prefix))
@@ -89,9 +87,7 @@ def test_if_empty_prefix_returns_everything(
         (get_tags, None, []),
     ],
 )
-def test_completion_of_nonexisting_prefix(
-    watson_df, func_to_test, rename_type, args
-):
+def test_completion_of_nonexisting_prefix(watson_df, func_to_test, rename_type, args):
     ctx = ClickContext(obj=watson_df, params={"rename_type": rename_type})
     prefix = "NOT-EXISTING-PREFIX"
     ret_list = list(func_to_test(ctx, args, prefix))