license

Author: MarcoGorelli

LICENSES/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2017 Anthony Sottile
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

pandas_style_guide/__main__.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 import tokenize
-
+import os
 
 import ast
 import importlib.metadata
@@ -14,8 +14,8 @@
 
 
 class Plugin:
-    name = __name__
-    version = 1#importlib.metadata.version(__name__)
+    name = os.path.split(os.path.dirname(__file__))[-1]
+    version = importlib.metadata.version(name)
 
     def __init__(self, tree: ast.AST):
         self._tree = tree

pandas_style_guide/_data.py

@@ -48,15 +48,15 @@ def visit(funcs, tree: ast.Module) -> Dict[int, List[int]]:
         for name in reversed(node._fields):
             value = getattr(node, name)
             if name in {"annotation", "returns"}:
-                next_in_annotation = True
+                next_state = state._replace(in_annotation=True)
             else:
-                next_in_annotation = state
+                next_state = state
             if isinstance(value, ast.AST):
-                nodes.append((next_in_annotation, value, node))
+                nodes.append((next_state, value, node))
             elif isinstance(value, list):
                 for value in reversed(value):
                     if isinstance(value, ast.AST):
-                        nodes.append((next_in_annotation, value, node))
+                        nodes.append((next_state, value, node))
 
 def _import_plugins() -> None:
     # https://github.com/python/mypy/issues/1422

pandas_style_guide/_plugins/namespace_usage.py

@@ -0,0 +1,10 @@
+
+import ast
+from pandas_style_guide._data import register
+from pandas_style_guide._ast_helpers import is_name_attr
+
+MSG = 'PSG010 namespace inconsistency'
+@register(ast.Attribute)
+def visit_Attribute(state, node: ast.Attribute, parent) -> None:
+    if node.attr in state.from_imports['pandas'] and isinstance(node.value, ast.Name) and node.value.id in {'pandas', 'pd'}:
+        yield node.lineno, node.col_offset, MSG

pandas_style_guide/_plugins/union.py

@@ -0,0 +1,21 @@
+
+import ast
+from pandas_style_guide._data import register
+from pandas_style_guide._ast_helpers import is_name_attr
+
+MSG = 'PSG010 union'
+
+@register(ast.Subscript)
+def visit_Subscript(state, node, parent):
+    if isinstance(node.value, ast.Name) and node.value.id == 'Union':
+        if isinstance(node.slice, ast.Index):
+            tuple = node.slice.value
+        else:
+            tuple = node.slice
+        names = set()
+        if isinstance(tuple, ast.Tuple):
+            for elt in tuple.elts:
+                if isinstance(elt, ast.Name):
+                    names.add(elt.id)
+        if names == {'DataFrame', 'Series'}:
+            yield node.lineno, node.col_offset, MSG

setup.cfg

@@ -1,11 +1,11 @@
 [metadata]
 name = pandas-style-guide
-version = 2.11.0
-description = A tool to automatically upgrade syntax for newer versions.
+version = 0.0.1
+description = Linter for contributing to pandas
 long_description = file: README.md
 long_description_content_type = text/markdown
-url = https://github.com/asottile/pyupgrade
-author = Anthony Sottile
+url = https://github.com/MarcoGorelli/pandas-style-guide
+author = Marco Gorelli
 author_email = [email protected]
 license = MIT
 license_file = LICENSE

t.py

@@ -1,8 +1,3 @@
-import pandas
-import conftest
-from pandas.foo.bar.dude import _foo
+from typing import Union
 
-_foo._foo('dod')
-os.remove('a')
-from os import remove
-remove('f')
+a: Union[DataFrame, Series]