55d37401bc
And no other module or component has an ImportError message for PIL. It is also outdated, as users might install pillow instead.
139 lines
4.3 KiB
Python
139 lines
4.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2010-2011 Jocelyn Jaubert
|
|
#
|
|
# This file is part of weboob.
|
|
#
|
|
# weboob is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# weboob is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import hashlib
|
|
|
|
from PIL import Image
|
|
|
|
from weboob.tools.log import getLogger
|
|
|
|
|
|
class TileError(Exception):
|
|
def __init__(self, msg, tile=None):
|
|
Exception.__init__(self, msg)
|
|
self.tile = tile
|
|
|
|
|
|
class Captcha(object):
|
|
def __init__(self, file, infos):
|
|
self.inim = Image.open(file)
|
|
self.infos = infos
|
|
self.nbr = int(infos["nbrows"])
|
|
self.nbc = int(infos["nbcols"])
|
|
(self.nx, self.ny) = self.inim.size
|
|
self.inmat = self.inim.load()
|
|
self.map = {}
|
|
|
|
self.tiles = [[Tile(y * self.nbc + x) for y in xrange(4)] for x in xrange(4)]
|
|
|
|
def __getitem__(self, coords):
|
|
x, y = coords
|
|
return self.inmat[x % self.nx, y % self.ny]
|
|
|
|
def all_coords(self):
|
|
for y in xrange(self.ny):
|
|
for x in xrange(self.nx):
|
|
yield x, y
|
|
|
|
def get_codes(self, code):
|
|
s = ''
|
|
num = 0
|
|
for c in code:
|
|
index = self.map[int(c)].id
|
|
keycode = str(self.infos["grid"][num * self.nbr * self.nbc + index])
|
|
s += keycode
|
|
if num < 5:
|
|
s += ','
|
|
num += 1
|
|
return s
|
|
|
|
def build_tiles(self):
|
|
for ty in xrange(0, self.nbc):
|
|
y = ty * 23
|
|
|
|
for tx in xrange(0, self.nbr):
|
|
x = tx * 24
|
|
|
|
tile = self.tiles[tx][ty]
|
|
|
|
for yy in xrange(y, y + 23):
|
|
for xx in xrange(x, x + 24):
|
|
tile.map.append(self[xx, yy])
|
|
|
|
num = tile.get_num()
|
|
if num > -1:
|
|
tile.valid = True
|
|
self.map[num] = tile
|
|
|
|
|
|
class Tile(object):
|
|
hash = {'ff1441b2c5f90703ef04e688e399aca5': 1,
|
|
'53d7f3dfd64f54723b231fc398b6be57': 2,
|
|
'5bcba7fa2107ba9a606e8d0131c162eb': 3,
|
|
'9db6e7ed063e5f9a69ab831e6cc0d721': 4,
|
|
'30ebb75bfa5077f41ccfb72e8c9cc15b': 5,
|
|
'61e27275e494038e524bc9fbbd0be130': 6,
|
|
'0e0816f1b743f320ca561f090df0fbb1': 7,
|
|
'11e7d4a6d447e66a5a112c1d9f7fc442': 8,
|
|
'2ea3c82768030d91571d360acf7a0f75': 9,
|
|
'28a834ebbf0238b46d3fffae1a0b781b': 0,
|
|
'04211db029ce488e07010f618a589c71': -1,
|
|
|
|
'9a1bdf493d4067e98d3f364586c81e9d': 1,
|
|
'932032493860463bb4a3df7c99a900ad': 2,
|
|
'59cd90f1fa0b416ecdb440bc16d0b8e7': 3,
|
|
'53fe822c5efebe5f6fdef0f272c29638': 4,
|
|
'2082a9c830c0c7c9c22e9c809c6cadf7': 5,
|
|
'7f24aa97f0037bddcf2a4c8c2dbf5948': 6,
|
|
'725b6f11f44ecc2e9f6e79e86e3a82a5': 7,
|
|
'61d57da23894b96fab11f7b83c055bba': 8,
|
|
'18f6290c1cfaecadc5992e7ef6047a49': 9,
|
|
'1ce77709ec1d7475685d7b50d6f1c89e': 0,
|
|
'6718858a509fff4b86604f3096cf65e1': -1,
|
|
}
|
|
|
|
def __init__(self, _id):
|
|
self.id = _id
|
|
self.valid = False
|
|
self.logger = getLogger('societegenerale.captcha')
|
|
self.map = []
|
|
|
|
def __repr__(self):
|
|
return "<Tile(%02d) valid=%s>" % (self.id, self.valid)
|
|
|
|
def checksum(self):
|
|
s = ''
|
|
for pxls in self.map:
|
|
for pxl in pxls:
|
|
s += '%02d' % pxl
|
|
return hashlib.md5(s).hexdigest()
|
|
|
|
def get_num(self):
|
|
sum = self.checksum()
|
|
try:
|
|
return self.hash[sum]
|
|
except KeyError:
|
|
self.display()
|
|
raise TileError('Tile not found ' + sum, self)
|
|
|
|
def display(self):
|
|
self.logger.debug(self.checksum())
|
|
#im = Image.new('RGB', (24, 23))
|
|
#im.putdata(self.map)
|
|
#im.save('/tmp/%s.png' % self.checksum())
|