parent
385b035a7c
commit
0973b98d45
|
@ -42,12 +42,8 @@ def process():
|
|||
return {"url": logic.process(link).secure}
|
||||
|
||||
|
||||
@json
|
||||
def error_handler(error):
|
||||
return {"error": str(error)}
|
||||
|
||||
for exc in EXCEPTIONS:
|
||||
app.register_error_handler(exc, error_handler)
|
||||
app.register_error_handler(exc, logic.error_handler)
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run()
|
||||
|
|
|
@ -28,6 +28,15 @@ class RedisContainer:
|
|||
self.name = name
|
||||
|
||||
|
||||
class RedisList(RedisContainer):
|
||||
""" Wrapper class for handling some use cases for
|
||||
redis lists as usual python list
|
||||
"""
|
||||
|
||||
def append(self, item):
|
||||
self._storage.lpush(self.name, item)
|
||||
|
||||
|
||||
class LinkSet(RedisContainer, Container):
|
||||
""" Wrapper class for handling some use cases for
|
||||
redis sets as usual python sets
|
||||
|
|
|
@ -3,6 +3,7 @@ from flask import current_app
|
|||
|
||||
from imgtohttps import storage
|
||||
from imgtohttps.lib import Link
|
||||
from imgtohttps.lib import json
|
||||
|
||||
|
||||
def has_secure_domain(link):
|
||||
|
@ -67,3 +68,9 @@ def process(link):
|
|||
else:
|
||||
result = upload(link)
|
||||
return result
|
||||
|
||||
@json
|
||||
def error_handler(error):
|
||||
err = str(error)
|
||||
storage.errors_list.append(err)
|
||||
return {"error": err}
|
|
@ -1,6 +1,6 @@
|
|||
from flask.ext.redis import FlaskRedis
|
||||
|
||||
from imgtohttps.lib import LinkSet, LinksMapping, LinkRegistry
|
||||
from imgtohttps.lib import LinkSet, LinksMapping, LinkRegistry, RedisList
|
||||
|
||||
|
||||
storage = FlaskRedis()
|
||||
|
@ -9,3 +9,4 @@ secure_domains = LinkSet(storage, 'secure_domains')
|
|||
insecure_domains = LinkSet(storage, 'insecure_domains')
|
||||
already_uploaded_links = LinksMapping(storage, 'already_uploaded_links')
|
||||
image_registry = LinkRegistry(storage, 'image_')
|
||||
errors_list = RedisList(storage, 'errors')
|
||||
|
|
Loading…
Reference in New Issue