pep8 stuff

Author: pltrdy

files2rouge/__init__.py

@@ -2,3 +2,5 @@
 from files2rouge.files2rouge import run
 from files2rouge.files2rouge import main
 from files2rouge.settings import Settings
+
+__all__ = [absolute_import, run, main, Settings]

files2rouge/files2rouge.py

@@ -43,26 +43,26 @@ def run(summ_path,
     print("Preparing documents...")
     utils.mkdirs([sys_root, model_root])
     utils.split_files(model_file=ref_path,
-                system_file=summ_path,
-                model_dir=model_root,
-                system_dir=sys_root,
-                eos=args.eos)
+                      system_file=summ_path,
+                      model_dir=model_root,
+                      system_dir=sys_root,
+                      eos=eos)
     print("Running ROUGE...")
     log_level = logging.ERROR if not verbose else None
     r = pyrouge.Rouge155(rouge_dir=os.path.dirname(s.data['ROUGE_path']),
                          log_level=log_level)
     r.system_dir = sys_root
     r.model_dir = model_root
-    r.system_filename_pattern = 's.(\d+).txt'
+    r.system_filename_pattern = r's.(\d+).txt'
     r.model_filename_pattern = 'm.[A-Z].#ID#.txt'
     data_arg = "-e %s" % s.data['ROUGE_data']
 
     if not rouge_args:
         rouge_args = [
-                  '-c', 95,
-                  '-r', 1000,
-                  '-n', 2,
-                  '-a']
+            '-c', 95,
+            '-r', 1000,
+            '-n', 2,
+            '-a']
         rouge_args_str = " ".join([str(_) for _ in rouge_args])
     else:
         rouge_args_str = rouge_args

files2rouge/settings.py

@@ -4,40 +4,42 @@
 PATHS = ['ROUGE_path', 'ROUGE_data']
 PARAMS = PATHS + []
 
+
 def _default_path():
-  _dir, _filename = os.path.split(__file__)
-  return os.path.join(_dir, 'settings.json')
+    _dir, _filename = os.path.split(__file__)
+    return os.path.join(_dir, 'settings.json')
 
 
 class Settings:
 
-  def __init__(self, path=None):
-    self.path = _default_path() if path is None else path
-
-  def _load(self):
-    try:
-        with open(self.path, 'r') as f:
-            data = json.load(f)
+    def __init__(self, path=None):
+        self.path = _default_path() if path is None else path
+
+    def _load(self):
+        try:
+            with open(self.path, 'r') as f:
+                data = json.load(f)
+            self.set_data(data)
+        except IOError:
+            print(
+                "Can't load ROUGE settings in '%s'. Check that the file "
+                "exists or initialize it with 'setup_rouge.py'" % self.path)
+            exit()
+
+    def _generate(self, data):
         self.set_data(data)
-    except IOError:
-        print("Can't load ROUGE settings in '%s'. Check that the file exists or initialize it with 'setup_rouge.py'" % self.path)
-        exit()
-
-
-  def _generate(self, data):
-    self.set_data(data)
-    with open(self.path, 'w') as f:
-      json.dump(data, f, indent=2)
-
-  def set_data(self, data):
-    """Check & set data to `data`
-    """
-    for param in PARAMS:
-      if param not in data:
-        raise ValueError('Missing parameter %d in data' % param)
-
-    for path_key in PATHS:
-      path = data[path_key]
-      if not os.path.exists(path):
-        raise ValueError("Path does not exist %s" % path)
-    self.data = data
+        with open(self.path, 'w') as f:
+            json.dump(data, f, indent=2)
+
+    def set_data(self, data):
+        """Check & set data to `data`
+        """
+        for param in PARAMS:
+            if param not in data:
+                raise ValueError('Missing parameter %d in data' % param)
+
+        for path_key in PATHS:
+            path = data[path_key]
+            if not os.path.exists(path):
+                raise ValueError("Path does not exist %s" % path)
+        self.data = data

files2rouge/utils.py

@@ -2,22 +2,26 @@
 from __future__ import print_function
 import os
 
+
 def mkdir(path):
     os.mkdir(path)
 
+
 def mkdirs(paths):
     for path in paths:
         mkdir(path)
 
+
 def tee(saveto, *args, **kwargs):
     """Mimic the tee command, write on both stdout and file
     """
     print(*args, **kwargs)
     if saveto is not None:
         print(file=saveto, *args, **kwargs)
 
+
 def split_files(model_file, system_file, model_dir, system_dir, eos="."):
-    def outputs(line, f): 
+    def outputs(line, f):
         split_sen = " .\n".join(line.split(" %s " % eos))
         print(split_sen, end="", file=f)
 
@@ -37,7 +41,6 @@ def outputs(line, f):
                 break
             if len(line) == 0:
                 continue
-        
+
             with open("%s/s.%d.txt" % (system_dir, i), "w") as f:
                 outputs(line, f)
-

setup.py

@@ -1,30 +1,32 @@
 from setuptools import setup, find_packages
 
-version="2.0.0"
+version = "2.0.0"
 setup(
     name="files2rouge",
     version=version,
     description="Calculating ROUGE score between two files (line-by-line)",
     url="http://github.com/pltrdy/files2rouge",
-    download_url="https://github.com/pltrdy/files2rouge/archive/%s.tar.gz" % version,
+    download_url="https://github.com/pltrdy/files2rouge/archive/%s.tar.gz"
+                 % version,
     author="pltrdy",
     author_email="pltrdy@gmail.com",
-    keywords=["NL", "CL", "natural language processing", "computational linguistics", "summarization"],
+    keywords=["NL", "CL", "natural language processing",
+              "computational linguistics", "summarization"],
     packages=find_packages(),
     classifiers=[
         "Intended Audience :: Science/Research",
         "Programming Language :: Python :: 3",
         "Topic :: Text Processing :: Linguistic"
-        ],
+    ],
     license="LICENCE.txt",
     long_description=open("README.md").read(),
 
     entry_points={
-      'console_scripts': [
-        'files2rouge=files2rouge:main'
-      ]
+        'console_scripts': [
+            'files2rouge=files2rouge:main'
+        ]
     },
-    install_requires = [
+    install_requires=[
     ],
     include_package_data=True,
 )

setup_rouge.py

@@ -4,48 +4,38 @@
     It has to be run before `setup.py`
 
 """
-from files2rouge import settings
 import os
 import shutil
-import sys
-
-
-def get_input(*args, **kwargs):
-    # Support Python 2 and 3 input
-    # Default to Python 3's input()
-    fct = input
 
-    # If this is Python 2, use raw_input()
-    if sys.version_info[:2] <= (2, 7):
-        fct = raw_input
-
-    return fct(*args, **kwargs)
+from files2rouge import settings
+from six.moves import input
 
-def copy_ROUGE():
-  home = os.environ['HOME']
 
-  src_ROUGE_root = "./files2rouge/RELEASE-1.5.5/"
+def copy_rouge():
+    home = os.environ['HOME']
 
-  default_root = os.path.join(home, '.files2rouge/')
+    src_rouge_root = "./files2rouge/RELEASE-1.5.5/"
 
-  print("files2rouge uses scripts and tools that will not be stored with the python package")
-  path = get_input("where do you want to save it? [default: %s]" % default_root)
+    default_root = os.path.join(home, '.files2rouge/')
 
-  if path == "": 
-    path = default_root
+    print("files2rouge uses scripts and tools that will not be stored with "
+          "the python package")
+    path = input(
+        "where do you want to save it? [default: %s]" % default_root)
 
-  ROUGE_data = os.path.join(path, "data")
-  ROUGE_path = os.path.join(path, "ROUGE-1.5.5.pl")
+    if path == "":
+        path = default_root
 
-  print("Copying '%s' to '%s'" % (src_ROUGE_root, path))
-  shutil.copytree(src_ROUGE_root, path) 
+    rouge_data = os.path.join(path, "data")
+    rouge_path = os.path.join(path, "ROUGE-1.5.5.pl")
 
+    print("Copying '%s' to '%s'" % (src_rouge_root, path))
+    shutil.copytree(src_rouge_root, path)
 
-  return {"ROUGE_path": ROUGE_path, "ROUGE_data": ROUGE_data}
+    return {"ROUGE_path": rouge_path, "ROUGE_data": rouge_data}
 
 
 conf_path = "./files2rouge/settings.json"
 s = settings.Settings(path=conf_path)
-data = copy_ROUGE()
+data = copy_rouge()
 s._generate(data)
-