Skip to content

Commit

Permalink
New examples
Browse files Browse the repository at this point in the history
  • Loading branch information
@BenjV
BenjV committed Nov 6, 2016
1 parent 42072b2 commit d8a5f80
Show file tree
Hide file tree
Showing 2 changed files with 258 additions and 277 deletions.
314 changes: 76 additions & 238 deletions ExamplePostProcess.py
View file
Original file line number Diff line number Diff line change
@@ -1,240 +1,78 @@
#This Script is an example of what is possible with the
#PostProcess feature of Auto-Sub
import sys
import os
import library.pythontwitter as twitter
import library.oauth2 as oauth
import library.pynma as pynma
import socket
import email.utils
import smtplib

from library.growl import gntp
from email.mime.text import MIMEText
#------------------------------------------------------------------------------------------------------------------
# This Script is an example of what is possible with the
# PostProcess feature of Auto-Sub
# It will expect the following arguments from autosub:
# argv[1]: Path + filename of the subtitle file
# argv[2]: Path + filename of the video file
# argv[3]: Language of the subtile
# argv[4]: Name of the Serie
# argv[5]: Season number of the serie
# argv[6]: Episode number of the serie
#
# This example moves the videofile and the subfile to a different location.
#------------------------------------------------------------------------------------------------------------------

import os, shutil, errno

#---------------------------------------#
# This is the Start of the main program #
#---------------------------------------#

# First we move the parameters we got from autosub to some easy reading names.
SubSpecs = sys.argv[1]
VideoSpecs = sys.argv[2]
Language = sys.argv[3]
SerieName = sys.argv[4]
SeasonNum = sys.argv[5]
EpisodeNum = sys.argv[6]

#-----------------------------------------------------------------------------------------------------------------#
# Put her the part of the foldername(SrtRoot) you want to be replaced with another folder(DstRooT. #
# Beware, Linux systems are case sensitive!!! #
# If you use a '\' character in a string you have to escape it with another '\' and use '\\' insted. #
#-----------------------------------------------------------------------------------------------------------------#
#SrtRoot = '/volume1/video/series'
#DstRoot = '/volume1/video/tv/shows'
SrtRoot = 'D:\sync\Tv Series'
DstRoot = 'D:\sync\TV\Series'

# here we normalise the path for the current OS
DstRoot = os.path.normpath(DstRoot)
SrtRoot = os.path.normpath(SrtRoot)
#-----------------------------------------------------------------------------------------------------------------#
# Here we create the destination folder.
# We catch the error and if it already exitst its fine otherwise we leave
#-----------------------------------------------------------------------------------------------------------------#

try:
from urlparse import parse_qsl
except:
from cgi import parse_qsl

# TODO: create notification library and integrate it

#OPTIONS EDIT HERE
#What options:
# twitter - send notifications to twitter
# growl - send notification to a growl server
# echo - echo's the arguments given to the script
# mail - send an email when a subtitle is downloaded
# nma - Notify Your Android

what = "echo"

#Growl Options:
growl_host = ""
growl_port = 23053
growl_pass = ""
#/Growl Options

#Twitter options:
token_key = ''
token_secret = ''
#/Twitter options

#Mail Options:
#If you don't need a password and username, set them to (leave them blank): ''
#Currently, only TLS is supported for encryption. If you don't need encryption, leave it blank.
#WARNING! YOUR PASSWORD IS STORED IN PLAIN TEXT MAKE SURE THE FILE IS NOT
#READABLE BY EVERYONE.
#!USE AT OWN RISK!
srv = 'smtp.gmail.com:587'
fromaddr = '[email protected]'
toaddrs = '[email protected]'
username = '[email protected]'
password = 'mysecretpassword'
subject = 'Subs info'
encryption = 'TLS'
#/Mail Options

#Notify My Android options:
nma_apikey = ''
#/Notify My Android options:

#/OPTIONS STOP EDITING BELOW THIS LINE
#MAIL APPLICATION SOURCE
def send_mail(message):
"""
Construct and send the email.
Keyword arguments:
message -- Full name of the subtitle
"""
msg = MIMEText('Hi,\n \n AutoSub downloaded the following subtitle:\n %s'\
% message)
msg['To'] = email.utils.formataddr(('Recipient', toaddrs))
msg['Subject'] = subject
msg = msg.as_string()

try:
server = smtplib.SMTP(srv)
if encryption == 'TLS':
server.starttls()
if username != '' and password != '':
server.login(username, password)
server.sendmail(fromaddr, toaddrs, msg)
server.quit()
print "Email notification send"
except:
print "ERROR: Unable to send an email notification. Check your email settings."


#GROWL APPLICATION SOURCE
def send_growl(host, port, message):
try:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect((host, port))
s.send(message)
response = gntp.parse_gntp(s.recv(1024))
s.close()
print "Growl notification send"
except socket.error:
print "ERROR: Unable to send growl notification to growl server. Check your settings."


def create_growl(message):
notice = gntp.GNTPNotice()
notice.add_header('Application-Name', "AutoSub")
notice.add_header('Notification-Name', "Subtitle Download")
notice.add_header('Notification-Title', "AutoSub: Subtitle Download")
notice.add_header('Notification-Text', message)
if growl_pass != "":
notice.set_password(growl_pass)
send_growl(growl_host, growl_port, notice.encode())


def register_growl():
register = gntp.GNTPRegister()
register.add_header('Application-Name', "AutoSub")
register.add_notification('Test', True)
register.add_notification('Subtitle Download', True)
if growl_pass != "":
register.set_password(growl_pass)
send_growl(growl_host, growl_port, register.encode())

notice = gntp.GNTPNotice()
notice.add_header('Application-Name', "AutoSub")
notice.add_header('Notification-Name', "Test")
notice.add_header('Notification-Title', "Testing notification")
notice.add_header('Notification-Text', "This is a test notification send by AutoSub, check!")
if growl_pass != "":
notice.set_password(growl_pass)
send_growl(growl_host, growl_port, notice.encode())

#TWITTER APPLICATION SOURCE
consumer_key = 'CRMvUogoJ5kMErtU9fiw'
consumer_secret = 'JqS5jJIWdokl5iijZmoBHNwRsknw7xmCxPggEsmo8'

REQUEST_TOKEN_URL = 'https://api.twitter.com/oauth/request_token'
ACCESS_TOKEN_URL = 'https://api.twitter.com/oauth/access_token'
AUTHORIZATION_URL = 'https://api.twitter.com/oauth/authorize'
SIGNIN_URL = 'https://api.twitter.com/oauth/authenticate'


def authorize_application():
consumer = oauth.Consumer(key=consumer_key, secret=consumer_secret)
oauth_client = oauth.Client(consumer)
response, content = oauth_client.request(REQUEST_TOKEN_URL, 'GET')

if response['status'] != '200':
print "ERROR: Invalid response"
else:
request_token = dict(parse_qsl(content))
print "Visit: " + AUTHORIZATION_URL + "?oauth_token=" + request_token['oauth_token']

token_key = request_token['oauth_token']
token_secret = request_token['oauth_token_secret']

token = oauth.Token(token_key, token_secret)
pin = raw_input('PIN: ').strip()
token.set_verifier(pin)

oauth_client2 = oauth.Client(consumer, token)
response, content = oauth_client2.request(ACCESS_TOKEN_URL, method='POST', body='oauth_verifier=%s' % pin)
access_token = dict(parse_qsl(content))

if response['status'] != '200':
print "ERROR: Invalid response"
else:
token_key = access_token['oauth_token']
token_secret = access_token['oauth_token_secret']
print "token_key='%s'" % token_key
print "token_secret='%s'" % token_secret


def send_tweet(message):
try:
api = twitter.Api(consumer_key, consumer_secret, token_key, token_secret)
api.PostUpdate(message)
print "Tweet sended!"
except:
print "ERROR: Could not send tweet, make sure token_key & token_secret are correct."

#Notify My Android source:
def notify_my_android(subtitle):
nma_instance = pynma.PyNMA(nma_apikey)
message = "Auto-Sub just downloaded the following subtitle \n %s" %subtitle

resp = nma_instance.push('Auto-Sub','Downloaded a Subtitle',message)
if not resp[nma_apikey][u'code'] == u'200':
print "ERROR: Something went wrong!"
else:
print "Android notified!"

#MAIN FUNCTION
if what == "echo":
print
try:
print sys.argv[1]
print sys.argv[2]
except IndexError:
print "ERROR: Trying to setup this script? Make sure you set the 'what' variable. "

if what == "twitter":
if token_key == "":
authorize_application()
else:
if (sys.argv[1] and sys.argv[2] and token_key != ""):
var = sys.argv[1]
var = os.path.basename(var)
tweet = "AutoSub Downloaded: " + var
print "Sending the following tweet %s" % tweet[:140]
send_tweet(tweet[:140])
elif (sys.argv[1] and sys.argv[2]):
print "ERROR: token_key not set yet, run script without any arguments"

if what == "growl":
if len(sys.argv) == 1:
register_growl()
elif (sys.argv[1] and sys.argv[2] and growl_host != ""):
var = sys.argv[1]
var = os.path.basename(var)
create_growl(var)

if what == "mail":
if (sys.argv[1] and sys.argv[2] and srv != "" and fromaddr != "" and toaddrs != ""):
var = sys.argv[1]
var = os.path.basename(var)
send_mail(var)
elif (sys.argv[1] and sys.argv[2]):
print "ERROR: Minimum mail settings aren't set."
else:
print "ERROR: Mail settings not set"

if what == "nma":
if (sys.argv[1] and sys.argv[2] and nma_apikey):
var = sys.argv[1]
var = os.path.basename(var)
notify_my_android(var)
elif (sys.argv[1] and sys.argv[2]):
print "ERROR: No APIKey set!"
else:
print "ERROR: Something went wrong!"
os.makedirs(DstRoot)
except OSError as exception:
if exception.errno != errno.EEXIST:
sys.stdout.write('- Could not create destination folder')
sys.exit(0)
#-----------------------------------------------------------------------------------------------------------------#
# Here we create the destination files specs bij replacing the first part of the spec with a new part
#-----------------------------------------------------------------------------------------------------------------#
DstSubSpecs = SubSpecs.replace(SrtRoot,DstRoot)

# Now we move the subtilefile, if we fail we leave with a error message
try:
shutil.move(SubSpecs,DstSubSpecs)
except Exception as error:
sys.stdout.write(error)
sys.exit(0)

#-----------------------------------------------------------------------------------------------------------------#
# Here we create the destination files specs bij replacing the first part of the spec with a new part
#-----------------------------------------------------------------------------------------------------------------#
DstVideoSpecs = VideoSpecs.replace(SrtRoot,DstRoot)
#-----------------------------------------------------------------------------------------------------------------#
# Now we move the subtilefile, if we fail we leave with a error message
#-----------------------------------------------------------------------------------------------------------------#
try:
shutil.move(VideoSpecs,DstVideoSpecs)
except Exception as error:
sys.stdout.write(error)
sys.exit(0)
sys.stdout.write(" - Postprocess routine finished succesfull")
sys.exit(0)
Loading

0 comments on commit d8a5f80

Please sign in to comment.