os remove

Author: MarcoGorelli

pandas_style_guide/__main__.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+import tokenize
 
 
 import ast
@@ -25,4 +26,4 @@ def run(self) -> Generator[tuple[int, int, str, type[Any]], None, None]:
             return
         for line, col, msg in callbacks:
             yield line, col, msg, type(self)
-                                                                  
+        

pandas_style_guide/_plugins/os_remove.py

@@ -0,0 +1,11 @@
+import ast
+from pandas_style_guide._data import register
+from pandas_style_guide._ast_helpers import is_name_attr
+
+MSG = 'PSG010 os remove'
+@register(ast.Call)
+def check_contextless_pytest_raises(state, node, parent):
+    if is_name_attr(node.func, state.from_imports, 'os', ('remove', )) and not isinstance(parent, ast.withitem):
+        yield node.lineno, node.col_offset, MSG
+    elif isinstance(node.func, ast.Attribute) and node.func.attr == 'remove' and isinstance(node.func.value, ast.Name) and node.func.value.id == 'os' and not isinstance(parent, ast.withitem):
+        yield node.lineno, node.col_offset, MSG

t.py

@@ -3,3 +3,6 @@
 from pandas.foo.bar.dude import _foo
 
 _foo._foo('dod')
+os.remove('a')
+from os import remove
+remove('f')