Skip to content

Commit 4f73917

Browse files
henryiiijpivarski
andauthored
chore: cleanup flake8 (#527)
* style: cleanup flake8 * style: more flake8 * style: clean a few more suggestions * fix: check for print statements * fix: restore a few more warnings * fix: update a few versions Signed-off-by: Henry Schreiner <[email protected]> * chore: ignore prints in dev Signed-off-by: Henry Schreiner <[email protected]> * tests: noqa and use == on None Signed-off-by: Henry Schreiner <[email protected]> * tests: direct falsiness for ROOT Signed-off-by: Henry Schreiner <[email protected]> Co-authored-by: Jim Pivarski <[email protected]>
1 parent 846b5cf commit 4f73917

File tree

7 files changed

+61
-59
lines changed

7 files changed

+61
-59
lines changed

.pre-commit-config.yaml

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,19 +33,24 @@ repos:
3333
rev: v1.20.1
3434
hooks:
3535
- id: setup-cfg-fmt
36-
args: ["--min-py3-version", "3.5"]
3736

3837
- repo: https://github.com/mgedmin/check-manifest
3938
rev: "0.48"
4039
hooks:
4140
- id: check-manifest
4241
stages: [manual]
4342

43+
- repo: https://github.com/asottile/yesqa
44+
rev: v1.3.0
45+
hooks:
46+
- id: yesqa
47+
additional_dependencies: &flake8-dependencies
48+
- flake8-bugbear==22.4.25
49+
- flake8-docstrings==1.6.0
50+
- flake8-print==5.0.0
51+
4452
- repo: https://github.com/PyCQA/flake8
4553
rev: 4.0.1
4654
hooks:
4755
- id: flake8
48-
additional_dependencies:
49-
- flake8-bugbear
50-
- flake8-docstrings
51-
- flake8-print
56+
additional_dependencies: *flake8-dependencies

dev/make-models.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,8 @@
6767

6868
for classname, class_version in superclasses:
6969
cls = f.file.class_named(classname, class_version)
70-
print(cls.class_code) # noqa: T201
71-
print( # noqa: T201
70+
print(cls.class_code)
71+
print(
7272
f"""
7373
writable = True
7474
@@ -93,8 +93,8 @@ class {uproot.model.classname_encode(classname)}(uproot.model.DispatchByVersion)
9393

9494
for key in keys:
9595
obj = f[key]
96-
print(type(obj).class_code) # noqa: T201
97-
print( # noqa: T201
96+
print(type(obj).class_code)
97+
print(
9898
"""
9999
class_rawstreamers = ("""
100100
)
@@ -108,15 +108,15 @@ class {uproot.model.classname_encode(classname)}(uproot.model.DispatchByVersion)
108108
).tobytes()
109109
preamble = b"\xff\xff\xff\xffTStreamerInfo\x00"
110110
full = header + preamble + inner + b"\x00"
111-
print( # noqa: T201
111+
print(
112112
f""" uproot._writing.RawStreamerInfo(
113113
None,
114114
{full},
115115
{streamer_name!r},
116116
{streamer_version},
117117
),"""
118118
)
119-
print( # noqa: T201
119+
print(
120120
f""" )
121121
writable = True
122122
@@ -132,7 +132,7 @@ def _serialize(self, out, header, name):
132132
"""
133133
)
134134

135-
print( # noqa: T201
135+
print(
136136
f"""
137137
class {uproot.model.classname_encode(obj.classname)}(uproot.model.DispatchByVersion):
138138
\"\"\"
@@ -144,12 +144,12 @@ class {uproot.model.classname_encode(obj.classname)}(uproot.model.DispatchByVers
144144
)
145145

146146
for classname, _ in superclasses:
147-
print( # noqa: T201
147+
print(
148148
f"uproot.classes[{classname!r}] = {uproot.model.classname_encode(classname)}"
149149
)
150150

151151
for key in keys:
152152
obj = f[key]
153-
print( # noqa: T201
153+
print(
154154
f"uproot.classes[{obj.classname!r}] = {uproot.model.classname_encode(obj.classname)}"
155155
)

setup.cfg

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,9 @@ profile = black
6464
multi_line_output = 3
6565

6666
[flake8]
67-
ignore = E203, W503, E501, E266, N, D200, D102, D105, D107, D401, D402, D205, D400, D301
67+
extend-ignore = E203, E501, E722, B950, E266, D
68+
extend-select = B9, E711, D103
6869
max-complexity = 100
6970
exclude = tests/*.py, setup.py, src/uproot/__init__.py, docs-sphinx/*.py
71+
per-file-ignores =
72+
dev/*: T

src/uproot/behaviors/TBranch.py

Lines changed: 14 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -895,7 +895,7 @@ def show(
895895
+ "\n"
896896
)
897897

898-
for name, branch in self.iteritems( # noqa: B301 (not a dict)
898+
for name, branch in self.iteritems(
899899
filter_name=filter_name,
900900
filter_typename=filter_typename,
901901
filter_branch=filter_branch,
@@ -1379,7 +1379,7 @@ def keys(
13791379
Returns the names of the subbranches as a list of strings.
13801380
"""
13811381
return list(
1382-
self.iterkeys( # noqa: B301 (not a dict)
1382+
self.iterkeys(
13831383
filter_name=filter_name,
13841384
filter_typename=filter_typename,
13851385
filter_branch=filter_branch,
@@ -1416,7 +1416,7 @@ def values(
14161416
:ref:`uproot.behaviors.TBranch.HasBranches.branches`.)
14171417
"""
14181418
return list(
1419-
self.itervalues( # noqa: B301 (not a dict)
1419+
self.itervalues(
14201420
filter_name=filter_name,
14211421
filter_typename=filter_typename,
14221422
filter_branch=filter_branch,
@@ -1453,7 +1453,7 @@ def items(
14531453
of (str, :doc:`uproot.behaviors.TBranch.TBranch`).
14541454
"""
14551455
return list(
1456-
self.iteritems( # noqa: B301 (not a dict)
1456+
self.iteritems(
14571457
filter_name=filter_name,
14581458
filter_typename=filter_typename,
14591459
filter_branch=filter_branch,
@@ -1527,7 +1527,7 @@ def iterkeys(
15271527
15281528
Returns the names of the subbranches as an iterator over strings.
15291529
"""
1530-
for k, _ in self.iteritems( # noqa: B301 (not a dict)
1530+
for k, _ in self.iteritems(
15311531
filter_name=filter_name,
15321532
filter_typename=filter_typename,
15331533
filter_branch=filter_branch,
@@ -1563,7 +1563,7 @@ def itervalues(
15631563
(Note: with ``recursive=False``, this is the same as
15641564
:ref:`uproot.behaviors.TBranch.HasBranches.branches`.)
15651565
"""
1566-
for _, v in self.iteritems( # noqa: B301 (not a dict)
1566+
for _, v in self.iteritems(
15671567
filter_name=filter_name,
15681568
filter_typename=filter_typename,
15691569
filter_branch=filter_branch,
@@ -1625,7 +1625,7 @@ def iteritems(
16251625
yield branch.name, branch
16261626

16271627
if recursive:
1628-
for k1, v in branch.iteritems( # noqa: B301 (not a dict)
1628+
for k1, v in branch.iteritems(
16291629
recursive=recursive,
16301630
filter_name=no_filter,
16311631
filter_typename=filter_typename,
@@ -1669,7 +1669,7 @@ def itertypenames(
16691669
Returns (name, typename) pairs of the subbranches as an iterator over
16701670
2-tuples of (str, str).
16711671
"""
1672-
for k, v in self.iteritems( # noqa: B301 (not a dict)
1672+
for k, v in self.iteritems(
16731673
filter_name=filter_name,
16741674
filter_typename=filter_typename,
16751675
filter_branch=filter_branch,
@@ -1682,7 +1682,7 @@ def _ipython_key_completions_(self):
16821682
"""
16831683
Supports key-completion in an IPython or Jupyter kernel.
16841684
"""
1685-
return self.iterkeys() # noqa: B301 (not a dict)
1685+
return self.iterkeys()
16861686

16871687
def num_entries_for(
16881688
self,
@@ -1800,7 +1800,7 @@ def common_entry_offsets(
18001800
:ref:`uproot.behaviors.TBranch.TBranch.entry_offsets`.
18011801
"""
18021802
common_offsets = None
1803-
for branch in self.itervalues( # noqa: B301 (not a dict)
1803+
for branch in self.itervalues(
18041804
filter_name=filter_name,
18051805
filter_typename=filter_typename,
18061806
filter_branch=filter_branch,
@@ -1834,9 +1834,7 @@ def __getitem__(self, where):
18341834

18351835
if "/" in where:
18361836
where = "/".join([x for x in where.split("/") if x != ""])
1837-
for k, v in self.iteritems( # noqa: B301 (not a dict)
1838-
recursive=True, full_paths=True
1839-
):
1837+
for k, v in self.iteritems(recursive=True, full_paths=True):
18401838
if where == k:
18411839
self._lookup[original_where] = v
18421840
return v
@@ -2787,7 +2785,7 @@ def _filter_name_deep(filter_name, hasbranches, branch):
27872785

27882786
def _keys_deep(hasbranches):
27892787
out = set()
2790-
for branch in hasbranches.itervalues(recursive=True): # noqa: B301 (not a dict)
2788+
for branch in hasbranches.itervalues(recursive=True):
27912789
name = branch.name
27922790
out.add(name)
27932791
while branch is not hasbranches:
@@ -3049,7 +3047,7 @@ def _regularize_expressions(
30493047
branchid_interpretation = {}
30503048

30513049
if expressions is None:
3052-
for branchname, branch in hasbranches.iteritems( # noqa: B301 (not a dict)
3050+
for branchname, branch in hasbranches.iteritems(
30533051
filter_name=filter_name,
30543052
filter_typename=filter_typename,
30553053
filter_branch=filter_branch,
@@ -3342,7 +3340,7 @@ def _hasbranches_num_entries_for(
33423340
hasbranches, target_num_bytes, entry_start, entry_stop, branchid_interpretation
33433341
):
33443342
total_bytes = 0.0
3345-
for branch in hasbranches.itervalues(recursive=True): # noqa: B301 (not a dict)
3343+
for branch in hasbranches.itervalues(recursive=True):
33463344
if branch.cache_key in branchid_interpretation:
33473345
entry_offsets = branch.entry_offsets
33483346
start = entry_offsets[0]

src/uproot/reading.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1620,7 +1620,7 @@ def keys(
16201620
Note that this does not read any data from the file.
16211621
"""
16221622
return list(
1623-
self.iterkeys( # noqa: B301 (not a dict)
1623+
self.iterkeys(
16241624
recursive=recursive,
16251625
cycle=cycle,
16261626
filter_name=filter_name,
@@ -1651,7 +1651,7 @@ def values(
16511651
and ``filter_classname``.
16521652
"""
16531653
return list(
1654-
self.itervalues( # noqa: B301 (not a dict)
1654+
self.itervalues(
16551655
recursive=recursive,
16561656
filter_name=filter_name,
16571657
filter_classname=filter_classname,
@@ -1683,7 +1683,7 @@ def items(
16831683
and ``filter_classname``.
16841684
"""
16851685
return list(
1686-
self.iteritems( # noqa: B301 (not a dict)
1686+
self.iteritems(
16871687
recursive=recursive,
16881688
cycle=cycle,
16891689
filter_name=filter_name,
@@ -1759,7 +1759,7 @@ def iterkeys(
17591759
seen.add(out)
17601760

17611761
if recursive and key.fClassName in ("TDirectory", "TDirectoryFile"):
1762-
for k1 in key.get().iterkeys( # noqa: B301 (not a dict)
1762+
for k1 in key.get().iterkeys(
17631763
recursive=recursive,
17641764
cycle=cycle,
17651765
filter_name=no_filter,
@@ -1794,7 +1794,7 @@ def itervalues(
17941794
Note that this reads all objects that are selected by ``filter_name``
17951795
and ``filter_classname``.
17961796
"""
1797-
for _, v in self.iteritems( # noqa: B301 (not a dict)
1797+
for _, v in self.iteritems(
17981798
recursive=recursive,
17991799
cycle=False,
18001800
filter_name=filter_name,
@@ -1839,7 +1839,7 @@ def iteritems(
18391839
seen.add(out)
18401840

18411841
if recursive and key.fClassName in ("TDirectory", "TDirectoryFile"):
1842-
for k1, v in key.get().iteritems( # noqa: B301 (not a dict)
1842+
for k1, v in key.get().iteritems(
18431843
recursive=recursive,
18441844
cycle=cycle,
18451845
filter_name=no_filter,
@@ -1905,7 +1905,7 @@ def _ipython_key_completions_(self):
19051905
"""
19061906
Supports key-completion in an IPython or Jupyter kernel.
19071907
"""
1908-
return self.iterkeys() # noqa: B301 (not a dict)
1908+
return self.iterkeys()
19091909

19101910
def __len__(self):
19111911
if self._len is None:
@@ -1925,7 +1925,7 @@ def __contains__(self, where):
19251925
return True
19261926

19271927
def __iter__(self):
1928-
return self.iterkeys() # noqa: B301 (not a dict)
1928+
return self.iterkeys()
19291929

19301930
def title_of(self, where):
19311931
"""

src/uproot/writing/writable.py

Lines changed: 9 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -601,13 +601,13 @@ def __contains__(self, where):
601601
return False
602602

603603
def __iter__(self):
604-
return self.iterkeys() # noqa B301 (not a dict)
604+
return self.iterkeys()
605605

606606
def _ipython_key_completions_(self):
607607
"""
608608
Supports key-completion in an IPython or Jupyter kernel.
609609
"""
610-
return self.iterkeys() # noqa: B301 (not a dict)
610+
return self.iterkeys()
611611

612612
def keys(
613613
self,
@@ -633,7 +633,7 @@ def keys(
633633
Note that this does not read any data from the file.
634634
"""
635635
return list(
636-
self.iterkeys( # noqa: B301 (not a dict)
636+
self.iterkeys(
637637
recursive=recursive,
638638
cycle=cycle,
639639
filter_name=filter_name,
@@ -664,7 +664,7 @@ def values(
664664
and ``filter_classname``.
665665
"""
666666
return list(
667-
self.itervalues( # noqa: B301 (not a dict)
667+
self.itervalues(
668668
recursive=recursive,
669669
filter_name=filter_name,
670670
filter_classname=filter_classname,
@@ -696,7 +696,7 @@ def items(
696696
and ``filter_classname``.
697697
"""
698698
return list(
699-
self.iteritems( # noqa: B301 (not a dict)
699+
self.iteritems(
700700
recursive=recursive,
701701
cycle=cycle,
702702
filter_name=filter_name,
@@ -771,9 +771,7 @@ def iterkeys(
771771
yield keyname
772772

773773
if recursive and classname in ("TDirectory", "TDirectoryFile"):
774-
for k1 in self._get( # noqa: B301 (not a dict)
775-
keyname, cyclenum
776-
).iterkeys(
774+
for k1 in self._get(keyname, cyclenum).iterkeys(
777775
recursive=recursive,
778776
cycle=cycle,
779777
filter_name=filter_name,
@@ -806,7 +804,7 @@ def itervalues(
806804
Note that this reads all objects that are selected by ``filter_name``
807805
and ``filter_classname``.
808806
"""
809-
for keyname in self.iterkeys( # noqa: B301 (not a dict)
807+
for keyname in self.iterkeys(
810808
recursive=recursive,
811809
cycle=True,
812810
filter_name=filter_name,
@@ -838,7 +836,7 @@ def iteritems(
838836
Note that this reads all objects that are selected by ``filter_name``
839837
and ``filter_classname``.
840838
"""
841-
for keyname in self.iterkeys( # noqa: B301 (not a dict)
839+
for keyname in self.iterkeys(
842840
recursive=recursive,
843841
cycle=True,
844842
filter_name=filter_name,
@@ -884,9 +882,7 @@ def iterclassnames(
884882
yield keyname, classname
885883

886884
if recursive and classname in ("TDirectory", "TDirectoryFile"):
887-
for k1, c1 in self._get(
888-
keyname, cyclenum
889-
).iterclassnames( # noqa: B301 (not a dict)
885+
for k1, c1 in self._get(keyname, cyclenum).iterclassnames(
890886
recursive=recursive,
891887
cycle=cycle,
892888
filter_name=filter_name,

0 commit comments

Comments
 (0)