fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-05-12 17:18:30 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improved mypy checks

Browse source
This commit is contained in:
Cédric Bonhomme 2019-12-15 21:51:39 +01:00
parent c0abda31bc
commit a61a7916a3
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
2 changed files with 15 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
stegano/steganalysis/statistics.py
View file

@ -25,6 +25,7 @@ __date__ = "$Date: 2010/10/01 $"
__revision__ = "$Date: 2016/08/26 $"
__license__ = "GPLv3"
import typing
import operator
from PIL import Image
@ -37,7 +38,7 @@ def steganalyse(img):
"""
encoded = img.copy()
width, height = img.size
colours_counter = Counter() # type: Counter[int]
colours_counter = Counter() # type: typing.Counter[int]
for row in range(height):
for col in range(width):
r, g, b = img.getpixel((col, row))

23
tools/run_mypy.py
View file

@ -1,18 +1,21 @@
import subprocess
import sys
modules = ["stegano/tools.py",
"stegano/lsb/lsb.py",
"stegano/lsbset/lsbset.py",
"stegano/lsbset/generators.py",
"stegano/red/red.py",
"stegano/exifHeader/exifHeader.py",
"stegano/steganalysis/parity.py",
"stegano/steganalysis/statistics.py"]
modules = ['stegano/tools.py',
'stegano/lsb/lsb.py',
'stegano/lsbset/lsbset.py',
'stegano/lsbset/generators.py',
'stegano/red/red.py',
'stegano/exifHeader/exifHeader.py',
'stegano/steganalysis/parity.py',
'stegano/steganalysis/statistics.py']
exit_codes = []
for module in modules:
rc = subprocess.call(["mypy", "--ignore-missing-imports",
"--follow-imports", "skip", module])
rc = subprocess.call(['mypy', '--ignore-missing-imports',
'--check-untyped-defs',
'--follow-imports', 'skip', module],
)
exit_codes.append(rc)
sys.exit(max(exit_codes))