88 lines
3.5 KiB
Python
88 lines
3.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2011 Laurent Bachelier
|
|
#
|
|
# 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/>.
|
|
|
|
|
|
|
|
|
|
from weboob.capabilities.paste import ICapPaste
|
|
import binascii
|
|
|
|
|
|
class BasePasteBackend(ICapPaste):
|
|
EXPIRATIONS = {}
|
|
"""
|
|
List of expirations and their corresponding remote codes (any type can be used).
|
|
The expirations, i.e. the keys, are integers representing the duration
|
|
in seconds. There also can be one False key, for the "forever" expiration.
|
|
"""
|
|
|
|
def get_closest_expiration(self, max_age):
|
|
"""
|
|
Get the expiration closest (and less or equal to) max_age (int, in seconds).
|
|
max_age set to False means we want it to never expire.
|
|
|
|
@return int or False if found, else None
|
|
"""
|
|
# "forever"
|
|
if max_age is False and False in self.EXPIRATIONS:
|
|
return max_age
|
|
# get timed expirations, longest first
|
|
expirations = sorted([e for e in self.EXPIRATIONS if e is not False], reverse=True)
|
|
# find the first expiration that is below or equal to the maximum wanted age
|
|
for e in expirations:
|
|
if max_age is False or max_age >= e:
|
|
return e
|
|
|
|
|
|
def image_mime(data_base64, supported_formats=('gif', 'jpeg', 'png')):
|
|
try:
|
|
beginning = data_base64[:24].decode('base64')
|
|
except binascii.Error, e:
|
|
return None
|
|
|
|
if 'gif' in supported_formats and 'GIF8' in beginning:
|
|
return 'image/gif'
|
|
elif 'jpeg' in supported_formats and 'JFIF' in beginning:
|
|
return 'image/jpeg'
|
|
elif 'png' in supported_formats and '\x89PNG' in beginning:
|
|
return 'image/png'
|
|
elif 'xcf' in supported_formats and 'gimp xcf' in beginning:
|
|
return 'image/x-xcf'
|
|
elif 'pdf' in supported_formats and '%PDF' in beginning:
|
|
return 'application/pdf'
|
|
elif 'tiff' in supported_formats and ('II\x00\x2a' in beginning or \
|
|
'MM\x2a\x00' in beginning):
|
|
return 'image/tiff'
|
|
|
|
def test():
|
|
class MockPasteBackend(BasePasteBackend):
|
|
def __init__(self, expirations):
|
|
self.EXPIRATIONS = expirations
|
|
|
|
# all expirations are too high
|
|
assert MockPasteBackend({1337: '', 42: '', False: ''}).get_closest_expiration(1) is None
|
|
# we found a suitable lower or equal expiration
|
|
assert MockPasteBackend({1337: '', 42: '', False: ''}).get_closest_expiration(84) is 42
|
|
assert MockPasteBackend({1337: '', 42: '', False: ''}).get_closest_expiration(False) is False
|
|
assert MockPasteBackend({1337: '', 42: ''}).get_closest_expiration(False) is 1337
|
|
assert MockPasteBackend({1337: '', 42: '', False: ''}).get_closest_expiration(1336) is 42
|
|
assert MockPasteBackend({1337: '', 42: '', False: ''}).get_closest_expiration(1337) is 1337
|
|
assert MockPasteBackend({1337: '', 42: '', False: ''}).get_closest_expiration(1338) is 1337
|
|
# this format should work, though of doubtful usage
|
|
assert MockPasteBackend([1337, 42, False]).get_closest_expiration(84) is 42
|