Merge pull request #61 from wizzro/master

niqdev · web-flow · commit 67182245afe5 · 2017-03-24T18:13:18.000Z
Add pushover notification
diff --git a/README.md b/README.md
@@ -244,6 +244,19 @@ Now you should be able to trigger the event
 python script/spider.py --config config/prod.cfg --notify join
 ```
 
+### Pushover notification
+
+* Get your [USER_KEY](https://pushover.net/)
+* Create a [new application](https://pushover.net/apps/build)
+* (Optional) Add an [icon](https://pushover.net/icons/9aqpv697p9g6wzo.png)
+* Change your pushover credentials in the config file
+
+```
+[pushover]
+pushover.user_key=PUSHOVER_USER_KEY
+pushover.api_key=PUSHOVER_API_KEY
+```
+
 ### Heroku
 
 Create a new branch
diff --git a/config/prod_example.cfg b/config/prod_example.cfg
@@ -60,3 +60,7 @@ join.api_key=API_KEY
 firebase.database_secret=FIREBASE_DATABASE_SECRET
 firebase.url=FIREBASE_URL
 firebase.path=/books
+
+[pushover]
+pushover.user_key=PUSHOVER_USER_KEY
+pushover.api_key=PUSHOVER_API_KEY
diff --git a/requirements.txt b/requirements.txt
@@ -13,4 +13,4 @@ paramiko==2.0.2
 cryptography==1.6
 scp==0.10.2
 onedrivesdk==1.1.8
-
+python-pushover==0.3
diff --git a/script/notification/mypushover.py b/script/notification/mypushover.py
@@ -0,0 +1,21 @@
+from logs import *
+import requests
+from pushover import Client
+
+class Pushover(object):
+    """
+    """
+
+    def __init__(self, config, packpub_info, upload_info):
+        self.__config = config
+        self.__packpub_info = packpub_info
+        self.__client = Client(self.__config.get('pushover', 'pushover.user_key'), api_token=self.__config.get('pushover', 'pushover.api_key'))
+
+
+    def send(self):
+        self.__client.send_message(self.__packpub_info['description'].encode('utf-8'), title="New book downloaded from Packt: " + self.__packpub_info['title'].encode('utf-8'), url="https://www.packtpub.com/packt/offers/free-learning", url_title="See more")
+        log_success('[+] notification sent to pushover')
+
+    def sendError(self, exception, source):
+        self.__client.send_message(repr(exception), title='packtpub-crawler {source}: Could not download ebook'.format(source=source))
+        log_success('[+] error notification sent to pushover')
diff --git a/script/notify.py b/script/notify.py
@@ -2,10 +2,12 @@
 from notification.ifttt import Ifttt
 from logs import *
 from notification.join import Join
+from notification.mypushover import Pushover
 
 SERVICE_GMAIL = 'gmail'
 SERVICE_IFTTT = 'ifttt'
 SERVICE_JOIN = 'join'
+SERVICE_PUSHOVER = 'pushover'
 
 class Notify(object):
     """
@@ -23,6 +25,8 @@ def __init__(self, config, packpub_info, upload_info, service_type):
             self.service = Ifttt(config, packpub_info, upload_info)
         elif service_type == SERVICE_JOIN:
             self.service = Join(config, packpub_info, upload_info)
+        elif service_type == SERVICE_PUSHOVER:
+            self.service = Pushover(config, packpub_info, upload_info)
 
     def run(self):
         """
diff --git a/script/spider.py b/script/spider.py
@@ -9,7 +9,7 @@
 from upload import Upload, SERVICE_GOOGLE_DRIVE, SERVICE_ONEDRIVE, SERVICE_DROPBOX, SERVICE_SCP
 from database import Database, DB_FIREBASE
 from logs import *
-from notify import Notify, SERVICE_GMAIL, SERVICE_IFTTT, SERVICE_JOIN
+from notify import Notify, SERVICE_GMAIL, SERVICE_IFTTT, SERVICE_JOIN, SERVICE_PUSHOVER
 from noBookException import NoBookException
 from alreadyClaimedException import AlreadyClaimedException
 
@@ -67,7 +67,7 @@ def main():
     parser.add_argument('-e', '--extras', action='store_true', help='download source code (if exists) and book cover')
     parser.add_argument('-u', '--upload', choices=[SERVICE_GOOGLE_DRIVE, SERVICE_ONEDRIVE, SERVICE_DROPBOX, SERVICE_SCP], help='upload to cloud')
     parser.add_argument('-a', '--archive', action='store_true', help='compress all file')
-    parser.add_argument('-n', '--notify', choices=[SERVICE_GMAIL, SERVICE_IFTTT, SERVICE_JOIN], help='notify after claim/download')
+    parser.add_argument('-n', '--notify', choices=[SERVICE_GMAIL, SERVICE_IFTTT, SERVICE_JOIN, SERVICE_PUSHOVER], help='notify after claim/download')
     parser.add_argument('-s', '--store', choices=[DB_FIREBASE], help='store info')
     parser.add_argument('-o', '--claimOnly', action='store_true', help='only claim books (no downloads/uploads)')
 
