122 строки
3.6 KiB
Python
122 строки
3.6 KiB
Python
import logging
|
|
import logging.config
|
|
import os
|
|
import posixpath
|
|
import re
|
|
import sys
|
|
|
|
import MySQLdb as mysql
|
|
import sqlalchemy.pool as pool
|
|
|
|
from six.moves.urllib.parse import urlencode
|
|
|
|
from services.settings import settings
|
|
|
|
import olympia.core.logger
|
|
|
|
# Ugh. But this avoids any olympia models or django imports at all.
|
|
# Perhaps we can import these without any problems and we can
|
|
# remove all this.
|
|
from olympia.constants.platforms import PLATFORMS
|
|
|
|
|
|
# This is not DRY: it's a copy of amo.helpers.user_media_path, to avoid an
|
|
# import (which should triggers an import loop).
|
|
# See bug 1055654.
|
|
def user_media_path(what):
|
|
"""Make it possible to override storage paths in settings.
|
|
|
|
By default, all storage paths are in the MEDIA_ROOT.
|
|
|
|
This is backwards compatible.
|
|
|
|
"""
|
|
default = os.path.join(settings.MEDIA_ROOT, what)
|
|
key = "{0}_PATH".format(what.upper())
|
|
return getattr(settings, key, default)
|
|
|
|
|
|
# This is not DRY: it's a copy of amo.helpers.user_media_url, to avoid an
|
|
# import (which should be avoided, according to the comments above, and which
|
|
# triggers an import loop).
|
|
# See bug 1055654.
|
|
def user_media_url(what):
|
|
"""
|
|
Generate default media url, and make possible to override it from
|
|
settings.
|
|
"""
|
|
default = '%s%s/' % (settings.MEDIA_URL, what)
|
|
key = "{0}_URL".format(what.upper().replace('-', '_'))
|
|
return getattr(settings, key, default)
|
|
|
|
|
|
PLATFORM_NAMES_TO_CONSTANTS = {
|
|
platform.api_name: platform.id for platform in PLATFORMS.values()
|
|
}
|
|
|
|
ADDON_SLUGS_UPDATE = {
|
|
1: 'extension',
|
|
2: 'theme',
|
|
3: 'extension',
|
|
4: 'search',
|
|
5: 'item',
|
|
6: 'extension',
|
|
7: 'plugin'}
|
|
|
|
|
|
version_re = re.compile(r"""(?P<major>\d+) # major (x in x.y)
|
|
\.(?P<minor1>\d+) # minor1 (y in x.y)
|
|
\.?(?P<minor2>\d+|\*)? # minor2 (z in x.y.z)
|
|
\.?(?P<minor3>\d+|\*)? # minor3 (w in x.y.z.w)
|
|
(?P<alpha>[a|b]?) # alpha/beta
|
|
(?P<alpha_ver>\d*) # alpha/beta version
|
|
(?P<pre>pre)? # pre release
|
|
(?P<pre_ver>\d)? # pre release version
|
|
""", re.VERBOSE)
|
|
|
|
|
|
def get_cdn_url(id, row):
|
|
host = user_media_url('addons')
|
|
url = posixpath.join(host, str(id), row['filename'])
|
|
params = urlencode({'filehash': row['hash']})
|
|
return '{0}?{1}'.format(url, params)
|
|
|
|
|
|
def getconn():
|
|
db = settings.SERVICES_DATABASE
|
|
return mysql.connect(host=db['HOST'], user=db['USER'],
|
|
passwd=db['PASSWORD'], db=db['NAME'], charset='utf8')
|
|
|
|
|
|
mypool = pool.QueuePool(getconn, max_overflow=10, pool_size=5, recycle=300)
|
|
|
|
|
|
def log_configure():
|
|
"""You have to call this to explicitly configure logging."""
|
|
cfg = {
|
|
'version': 1,
|
|
'filters': {},
|
|
'handlers': {
|
|
'mozlog': {
|
|
'level': 'DEBUG',
|
|
'class': 'logging.StreamHandler',
|
|
'formatter': 'json'
|
|
},
|
|
},
|
|
'formatters': {
|
|
'json': {
|
|
'()': olympia.core.logger.JsonFormatter,
|
|
'logger_name': 'http_app_addons'
|
|
},
|
|
},
|
|
}
|
|
logging.config.dictConfig(cfg)
|
|
|
|
|
|
def log_exception(data):
|
|
# Note: although this logs exceptions, it logs at the info level so that
|
|
# on prod, we log at the error level and result in no logs on prod.
|
|
typ, value, discard = sys.exc_info()
|
|
error_log = olympia.core.logger.getLogger('z.update')
|
|
error_log.exception(u'Type: %s, %s. Data: %s' % (typ, value, data))
|