mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2024-11-14 17:40:24 -08:00
0acf78f196
Added migration code to migrate SickBeard section with fork sickrage-api to new SiCKRAGE section
95 lines
2.4 KiB
Python
95 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
oauthlib.utils
|
|
~~~~~~~~~~~~~~
|
|
|
|
This module contains utility methods used by various parts of the OAuth 2 spec.
|
|
"""
|
|
from __future__ import absolute_import, unicode_literals
|
|
|
|
import datetime
|
|
import os
|
|
|
|
from oauthlib.common import unicode_type, urldecode
|
|
|
|
try:
|
|
from urllib import quote
|
|
except ImportError:
|
|
from urllib.parse import quote
|
|
try:
|
|
from urlparse import urlparse
|
|
except ImportError:
|
|
from urllib.parse import urlparse
|
|
|
|
|
|
def list_to_scope(scope):
|
|
"""Convert a list of scopes to a space separated string."""
|
|
if isinstance(scope, unicode_type) or scope is None:
|
|
return scope
|
|
elif isinstance(scope, (set, tuple, list)):
|
|
return " ".join([unicode_type(s) for s in scope])
|
|
else:
|
|
raise ValueError("Invalid scope (%s), must be string, tuple, set, or list." % scope)
|
|
|
|
|
|
def scope_to_list(scope):
|
|
"""Convert a space separated string to a list of scopes."""
|
|
if isinstance(scope, (tuple, list, set)):
|
|
return [unicode_type(s) for s in scope]
|
|
elif scope is None:
|
|
return None
|
|
else:
|
|
return scope.strip().split(" ")
|
|
|
|
|
|
def params_from_uri(uri):
|
|
params = dict(urldecode(urlparse(uri).query))
|
|
if 'scope' in params:
|
|
params['scope'] = scope_to_list(params['scope'])
|
|
return params
|
|
|
|
|
|
def host_from_uri(uri):
|
|
"""Extract hostname and port from URI.
|
|
|
|
Will use default port for HTTP and HTTPS if none is present in the URI.
|
|
"""
|
|
default_ports = {
|
|
'HTTP': '80',
|
|
'HTTPS': '443',
|
|
}
|
|
|
|
sch, netloc, path, par, query, fra = urlparse(uri)
|
|
if ':' in netloc:
|
|
netloc, port = netloc.split(':', 1)
|
|
else:
|
|
port = default_ports.get(sch.upper())
|
|
|
|
return netloc, port
|
|
|
|
|
|
def escape(u):
|
|
"""Escape a string in an OAuth-compatible fashion.
|
|
|
|
TODO: verify whether this can in fact be used for OAuth 2
|
|
|
|
"""
|
|
if not isinstance(u, unicode_type):
|
|
raise ValueError('Only unicode objects are escapable.')
|
|
return quote(u.encode('utf-8'), safe=b'~')
|
|
|
|
|
|
def generate_age(issue_time):
|
|
"""Generate a age parameter for MAC authentication draft 00."""
|
|
td = datetime.datetime.now() - issue_time
|
|
age = (td.microseconds + (td.seconds + td.days * 24 * 3600)
|
|
* 10 ** 6) / 10 ** 6
|
|
return unicode_type(age)
|
|
|
|
|
|
def is_secure_transport(uri):
|
|
"""Check if the uri is over ssl."""
|
|
if os.environ.get('OAUTHLIB_INSECURE_TRANSPORT'):
|
|
return True
|
|
return uri.lower().startswith('https://')
|