Ruud
8 years ago
19 changed files with 547 additions and 121 deletions
@ -0,0 +1,84 @@ |
|||||
|
from couchpotato.core.helpers.encoding import toUnicode |
||||
|
from couchpotato.core.helpers.encoding import tryUrlencode |
||||
|
from couchpotato.core.helpers.variable import splitString |
||||
|
from couchpotato.core.logger import CPLog |
||||
|
from couchpotato.core.notifications.base import Notification |
||||
|
|
||||
|
|
||||
|
log = CPLog(__name__) |
||||
|
|
||||
|
autoload = 'Join' |
||||
|
|
||||
|
|
||||
|
class Join(Notification): |
||||
|
|
||||
|
# URL for request |
||||
|
url = 'https://joinjoaomgcd.appspot.com/_ah/api/messaging/v1/sendPush?title=%s&text=%s&deviceId=%s&icon=%s' |
||||
|
|
||||
|
# URL for notification icon |
||||
|
icon = tryUrlencode('https://raw.githubusercontent.com/CouchPotato/CouchPotatoServer/master/couchpotato/static/images/icons/android.png') |
||||
|
|
||||
|
def notify(self, message = '', data = None, listener = None): |
||||
|
if not data: data = {} |
||||
|
|
||||
|
# default for devices |
||||
|
device_default = [None] |
||||
|
|
||||
|
apikey = self.conf('apikey') |
||||
|
if apikey is not None: |
||||
|
# Add apikey to request url |
||||
|
self.url = self.url + '&apikey=' + apikey |
||||
|
# If api key is present, default to sending to all devices |
||||
|
device_default = ['group.all'] |
||||
|
|
||||
|
devices = self.getDevices() or device_default |
||||
|
successful = 0 |
||||
|
for device in devices: |
||||
|
response = self.urlopen(self.url % (self.default_title, tryUrlencode(toUnicode(message)), device, self.icon)) |
||||
|
|
||||
|
if response: |
||||
|
successful += 1 |
||||
|
else: |
||||
|
log.error('Unable to push notification to Join device with ID %s' % device) |
||||
|
|
||||
|
return successful == len(devices) |
||||
|
|
||||
|
def getDevices(self): |
||||
|
return splitString(self.conf('devices')) |
||||
|
|
||||
|
|
||||
|
config = [{ |
||||
|
'name': 'join', |
||||
|
'groups': [ |
||||
|
{ |
||||
|
'tab': 'notifications', |
||||
|
'list': 'notification_providers', |
||||
|
'name': 'join', |
||||
|
'options': [ |
||||
|
{ |
||||
|
'name': 'enabled', |
||||
|
'default': 0, |
||||
|
'type': 'enabler', |
||||
|
}, |
||||
|
{ |
||||
|
'name': 'devices', |
||||
|
'default': '', |
||||
|
'description': 'IDs of devices to notify, or group to send to if API key is specified (ex: group.all)' |
||||
|
}, |
||||
|
{ |
||||
|
'name': 'apikey', |
||||
|
'default': '', |
||||
|
'advanced': True, |
||||
|
'description': 'API Key for sending to all devices, or group' |
||||
|
}, |
||||
|
{ |
||||
|
'name': 'on_snatch', |
||||
|
'default': 0, |
||||
|
'type': 'bool', |
||||
|
'advanced': True, |
||||
|
'description': 'Also send message when movie is snatched.', |
||||
|
}, |
||||
|
], |
||||
|
} |
||||
|
], |
||||
|
}] |
||||
@ -0,0 +1,190 @@ |
|||||
|
import os |
||||
|
import base64 |
||||
|
import logging |
||||
|
import argparse |
||||
|
|
||||
|
import requests |
||||
|
|
||||
|
LOG_LEVEL = logging.INFO |
||||
|
DEFAULT_CHUNK_SIZE = 4 * 1024 * 1024 |
||||
|
TUS_VERSION = '1.0.0' |
||||
|
|
||||
|
logger = logging.getLogger(__name__) |
||||
|
logger.setLevel(logging.DEBUG) |
||||
|
logger.addHandler(logging.NullHandler()) |
||||
|
|
||||
|
|
||||
|
class TusError(Exception): |
||||
|
pass |
||||
|
|
||||
|
|
||||
|
def _init(): |
||||
|
fmt = "[%(asctime)s] %(levelname)s %(message)s" |
||||
|
h = logging.StreamHandler() |
||||
|
h.setLevel(LOG_LEVEL) |
||||
|
h.setFormatter(logging.Formatter(fmt)) |
||||
|
logger.addHandler(h) |
||||
|
|
||||
|
|
||||
|
def _create_parser(): |
||||
|
parser = argparse.ArgumentParser() |
||||
|
parser.add_argument('file', type=argparse.FileType('rb')) |
||||
|
parser.add_argument('--chunk-size', type=int, default=DEFAULT_CHUNK_SIZE) |
||||
|
parser.add_argument( |
||||
|
'--header', |
||||
|
action='append', |
||||
|
help="A single key/value pair" |
||||
|
" to be sent with all requests as HTTP header." |
||||
|
" Can be specified multiple times to send more then one header." |
||||
|
" Key and value must be separated with \":\".") |
||||
|
return parser |
||||
|
|
||||
|
|
||||
|
def _cmd_upload(): |
||||
|
_init() |
||||
|
|
||||
|
parser = _create_parser() |
||||
|
parser.add_argument('tus_endpoint') |
||||
|
parser.add_argument('--file_name') |
||||
|
parser.add_argument( |
||||
|
'--metadata', |
||||
|
action='append', |
||||
|
help="A single key/value pair to be sent in Upload-Metadata header." |
||||
|
" Can be specified multiple times to send more than one pair." |
||||
|
" Key and value must be separated with space.") |
||||
|
args = parser.parse_args() |
||||
|
|
||||
|
headers = dict([x.split(':') for x in args.header]) |
||||
|
metadata = dict([x.split(' ') for x in args.metadata]) |
||||
|
|
||||
|
upload( |
||||
|
args.file, |
||||
|
args.tus_endpoint, |
||||
|
chunk_size=args.chunk_size, |
||||
|
file_name=args.file_name, |
||||
|
headers=headers, |
||||
|
metadata=metadata) |
||||
|
|
||||
|
|
||||
|
def _cmd_resume(): |
||||
|
_init() |
||||
|
|
||||
|
parser = _create_parser() |
||||
|
parser.add_argument('file_endpoint') |
||||
|
args = parser.parse_args() |
||||
|
|
||||
|
headers = dict([x.split(':') for x in args.header]) |
||||
|
|
||||
|
resume( |
||||
|
args.file, |
||||
|
args.file_endpoint, |
||||
|
chunk_size=args.chunk_size, |
||||
|
headers=headers) |
||||
|
|
||||
|
|
||||
|
def upload(file_obj, |
||||
|
tus_endpoint, |
||||
|
chunk_size=DEFAULT_CHUNK_SIZE, |
||||
|
file_name=None, |
||||
|
headers=None, |
||||
|
metadata=None): |
||||
|
file_name = os.path.basename(file_obj.name) |
||||
|
file_size = _get_file_size(file_obj) |
||||
|
location = _create_file( |
||||
|
tus_endpoint, |
||||
|
file_name, |
||||
|
file_size, |
||||
|
extra_headers=headers, |
||||
|
metadata=metadata) |
||||
|
resume( |
||||
|
file_obj, location, chunk_size=chunk_size, headers=headers, offset=0) |
||||
|
|
||||
|
|
||||
|
def _get_file_size(f): |
||||
|
pos = f.tell() |
||||
|
f.seek(0, 2) |
||||
|
size = f.tell() |
||||
|
f.seek(pos) |
||||
|
return size |
||||
|
|
||||
|
|
||||
|
def _create_file(tus_endpoint, |
||||
|
file_name, |
||||
|
file_size, |
||||
|
extra_headers=None, |
||||
|
metadata=None): |
||||
|
logger.info("Creating file endpoint") |
||||
|
|
||||
|
headers = { |
||||
|
"Tus-Resumable": TUS_VERSION, |
||||
|
"Upload-Length": str(file_size), |
||||
|
} |
||||
|
|
||||
|
if extra_headers: |
||||
|
headers.update(extra_headers) |
||||
|
|
||||
|
if metadata: |
||||
|
l = [k + ' ' + base64.b64encode(v) for k, v in metadata.items()] |
||||
|
headers["Upload-Metadata"] = ','.join(l) |
||||
|
|
||||
|
response = requests.post(tus_endpoint, headers=headers) |
||||
|
if response.status_code != 201: |
||||
|
raise TusError("Create failed: %s" % response) |
||||
|
|
||||
|
location = response.headers["Location"] |
||||
|
logger.info("Created: %s", location) |
||||
|
return location |
||||
|
|
||||
|
|
||||
|
def resume(file_obj, |
||||
|
file_endpoint, |
||||
|
chunk_size=DEFAULT_CHUNK_SIZE, |
||||
|
headers=None, |
||||
|
offset=None): |
||||
|
if offset is None: |
||||
|
offset = _get_offset(file_endpoint, extra_headers=headers) |
||||
|
|
||||
|
total_sent = 0 |
||||
|
file_size = _get_file_size(file_obj) |
||||
|
while offset < file_size: |
||||
|
file_obj.seek(offset) |
||||
|
data = file_obj.read(chunk_size) |
||||
|
offset = _upload_chunk( |
||||
|
data, offset, file_endpoint, extra_headers=headers) |
||||
|
total_sent += len(data) |
||||
|
logger.info("Total bytes sent: %i", total_sent) |
||||
|
|
||||
|
|
||||
|
def _get_offset(file_endpoint, extra_headers=None): |
||||
|
logger.info("Getting offset") |
||||
|
|
||||
|
headers = {"Tus-Resumable": TUS_VERSION} |
||||
|
|
||||
|
if extra_headers: |
||||
|
headers.update(extra_headers) |
||||
|
|
||||
|
response = requests.head(file_endpoint, headers=headers) |
||||
|
response.raise_for_status() |
||||
|
|
||||
|
offset = int(response.headers["Upload-Offset"]) |
||||
|
logger.info("offset=%i", offset) |
||||
|
return offset |
||||
|
|
||||
|
|
||||
|
def _upload_chunk(data, offset, file_endpoint, extra_headers=None): |
||||
|
logger.info("Uploading chunk from offset: %i", offset) |
||||
|
|
||||
|
headers = { |
||||
|
'Content-Type': 'application/offset+octet-stream', |
||||
|
'Upload-Offset': str(offset), |
||||
|
'Tus-Resumable': TUS_VERSION, |
||||
|
} |
||||
|
|
||||
|
if extra_headers: |
||||
|
headers.update(extra_headers) |
||||
|
|
||||
|
response = requests.patch(file_endpoint, headers=headers, data=data) |
||||
|
if response.status_code != 204: |
||||
|
raise TusError("Upload chunk failed: %s" % response) |
||||
|
|
||||
|
return int(response.headers["Upload-Offset"]) |
||||
Loading…
Reference in new issue