See and Modify the name of a group the bot is in

README.md

@@ -1,7 +1,11 @@
-# SnapchatBot: Python library for building bots that live on Snapchat
+<h1 style='color: red'>This repo is deprecated due to changes in Snapchat's unofficial API.</h1>
+
+![SnapchatBot](http://i.imgur.com/s8XADUn.png?1)
+
+# SnapchatBot: A library for making bots that live on Snapchat
 
 Introducing SnapchatBot, an easy way to program Snapchat accounts to do anything you want.
-SnapchatBot can be used to easily create image-based notification services, chatbots, search interfaces,
+SnapchatBot can be used to create image-based notification services, chatbots, search interfaces,
 and any kind of intelligent agent that uses picture messages as its interaction mode.
 
 ## Bots Included
@@ -15,7 +19,13 @@ Sends back everything you send it.
 *(source at examples/storifierbot.py)*
 
 Takes all the snaps sent to it and adds them to its story. It can be used to collect responses
-from multiple people around a single theme, much like a Twitter hashtag.
+from multiple people around a single theme, much like a Twitter
+hashtag.
+
+#### The Capture Bot (by [EthanBlackburn](https://github.com/EthanBlackburn))
+*(source at examples/capturebot.py)*
+
+Saves all snaps received to the current working directory.
 
 #### The Auto-Welcomer Bot
 *(source at examples/autowelcomebot.py)*
@@ -42,11 +52,16 @@ Posts popular GIFs taken from the [Giphy](http://giphy.com) home page to its sto
 
 When you add the Connector to your friends, it links you with a stranger who's also added it. Every snap sent to the Connector will then arrive at the stranger's inbox, and all snaps sent from the stranger to the Connector will come to you. It's like ChatRoulette on Snapchat.
 
+#### The Rando Bot (by [PhlexPlexico](https://github.com/Phlexplexico))
+*(source at examples/randobot.py)*
+
+When you add RandoBot to your friends, it throws you in a list of people who've also added it. When you send it a snap, it will send your snap to a random person in the list. Similar to the [Rando](http://techcrunch.com/2014/03/22/rip-rando/) app.
+
 ## Installation
 
     $ python setup.py install
 
-You also need to have [ffmpeg](https://www.ffmpeg.org/) and [ImageMagick](http://www.imagemagick.org/) installed.
+You also need to have [ffmpeg](https://www.ffmpeg.org/), [ImageMagick](http://www.imagemagick.org/), and [libjpeg](http://libjpeg.sourceforge.net/) installed.
 
 ## How to build your own bots
 
@@ -57,6 +72,9 @@ You also need to have [ffmpeg](https://www.ffmpeg.org/) and [ImageMagick](http:/
 * `SnapchatBot#delete_friend(username)` -- deletes user with username `username` from the bot's friends
 * `SnapchatBot#block(username)` -- blocks user with username `username`
 * `SnapchatBot#get_snaps(mark_viewed = True)` -- gets snaps in the bot's inbox that haven't been viewed yet (use `mark_viewed = False` as a keyword argument if you don't want the bot to mark every snap received as viewed)
+* `SnapchatBot#get_my_stories()` -- gets all snaps in the bot's story
+* `SnapchatBot#get_friend_stories()` -- gets all the stories of the bot's friends
+* `SnapchatBot#clear_stories()` -- deletes all the bot's stories
 * `SnapchatBot#mark_viewed(snap)` -- marks `snap` as viewed
 * `SnapchatBot#get_friends()` -- gets the bot's friends
 * `SnapchatBot#get_added_me()` -- gets all users that have added the bot to their friends

examples/capturebot.py

@@ -0,0 +1,16 @@
+from argparse import ArgumentParser
+from snapchat_bots import SnapchatBot
+
+class CaptureBot(SnapchatBot):
+    def on_snap(self, sender, snap):
+        snap.save()
+
+if __name__ == '__main__':
+    parser = ArgumentParser("Capture Bot")
+    parser.add_argument('-u', '--username', required=True, type=str, help="Username of the account to run the bot on")
+    parser.add_argument('-p', '--password', required=True, type=str, help="Password of the account to run the bot on")
+
+    args = parser.parse_args()
+
+    bot = CaptureBot(args.username, args.password)
+    bot.listen(timeout=60)

examples/randobot.py

@@ -0,0 +1,60 @@
+from argparse import ArgumentParser
+from snapchat_bots import SnapchatBot, Snap
+import random
+
+class RandoBot(SnapchatBot):
+    def initialize(self):
+        self.connections = self.get_friends()
+	#If your bot ever gets blocked, uncomment these lines.
+	#Of course, make sure you have your old users backed up
+	#to the users.txt file! So you must uncomment the first
+	#three lines, while logged into the blocked bot, then
+	#uncomment the rest to re-add all users from the old bot.
+	#with open('users.txt', 'w') as file:
+	#	for item in self.connections:
+	#    		print>>file, item
+	#f = open('users.txt', 'r')
+	#for line in f:
+	#	self.add_friend(line)
+        #	print(line)
+	print(self.connections)
+
+    def connect(self,user):
+        self.log("Added user: %s to the array!" % (user))
+        self.connections.append(user)
+
+    def on_friend_add(self,friend):
+        self.add_friend(friend)
+        self.connect(friend)
+
+    def on_friend_delete(self,friend):
+        self.delete_friend(friend)
+        self.connections.remove(friend)
+
+    def find_random_user(self,username):
+	if len(self.connections) <= 1:
+		return None
+        newuser = random.choice(self.connections)
+	while(newuser == username):
+		newuser = random.choice(self.connections)
+        return newuser
+
+    def on_snap(self,sender,snap):
+        connection = self.find_random_user(sender)
+	if sender not in self.connections:
+		self.send_snap([sender], Snap.from_file("../resources/rando_addme.png"))
+    	if connection:
+		self.send_snap([connection],snap)
+		print("%s sent  snap to %s" % (sender,[connection]))
+	else:
+		self.send_snap([sender], Snap.from_file("../resources/rando_welcome.png"))
+
+if __name__ == '__main__':
+    parser = ArgumentParser("RandoBot Bot")
+    parser.add_argument('-u', '--username', required=True, type=str, help="Username of the account to run the bot on")
+    parser.add_argument('-p', '--password', required=True, type=str, help="Password of the account to run the bot on")
+
+    args = parser.parse_args()
+
+    bot = RandoBot(args.username, args.password)
+    bot.listen(timeout=33)

setup.py

@@ -16,8 +16,8 @@
     packages=['snapchat_bots'],
     install_requires=[
         'schedule>=0.3.1',
-        'requests>=2.5.1',
-        'PIL>=1.1.7',
+        'requests>=2.5.3',
+        'Pillow>=2.7.0',
         'pysnap>=0.1.1'
     ],
     dependency_links = ['https://github.com/martinp/pysnap/tarball/master#egg=pysnap-0.1.1'],

snapchat_bots/bot.py

@@ -1,7 +1,8 @@
-import logging, time, uuid
+import logging, time, uuid, requests, base64
 from pysnap import Snapchat
+from pysnap.utils import make_request_token, timestamp
 from snap import Snap
-from constants import DEFAULT_TIMEOUT
+from constants import DEFAULT_TIMEOUT, STATIC_TOKEN, BASE_URL
 
 FORMAT = '[%(asctime)-15s] %(message)s'
 logging.basicConfig(format=FORMAT)
@@ -12,11 +13,22 @@ class SnapchatBot(object):
     def __init__(self, username, password, **kwargs):
         self.bot_id = uuid.uuid4().hex[0:4]
 
+        self.auth_token = STATIC_TOKEN
+
         self.username = username
         self.password = password
 
+        r = self._make_request("/loq/login", {
+            'username': self.username,
+            'password': self.password
+        })
+
+        result = r.json()
+        self.auth_token = result['updates_response']['auth_token']
+
         self.client = Snapchat()
-        self.client.login(username, password)
+        self.client.username = username
+        self.client.auth_token = self.auth_token
 
         self.current_friends = self.get_friends()
         self.added_me = self.get_added_me()
@@ -28,7 +40,7 @@ def log(self, message, level=logging.DEBUG):
         logger.log(level, "[%s-%s] %s" % (self.__class__.__name__, self.bot_id, message))
 
     @staticmethod
-    def process_snap(snap_obj, data):
+    def process_snap(snap_obj, data, is_story = False):
         media_type = snap_obj["media_type"]
         sender = snap_obj["sender"]
         snap_id = snap_obj['id']
@@ -37,7 +49,8 @@ def process_snap(snap_obj, data):
                     snap_id=snap_id,
                     media_type=media_type,
                     duration=duration,
-                    sender=sender)
+                    sender=sender,
+                    is_story=is_story)
         return snap
 
     def mark_viewed(self, snap):
@@ -79,11 +92,7 @@ def get_added_me(self):
         return map(lambda fr: fr['name'], updates["added_friends"])
 
     def send_snap(self, recipients, snap):
-        self.log("Preparing to send snap %s" % snap.snap_id)
-
-        if not snap.uploaded:
-            self.log("Uploading snap %s" % snap.snap_id)
-            snap.upload(self)
+        media_id = self._upload_snap(snap)
 
         if type(recipients) is not list:
             recipients = [recipients]
@@ -92,15 +101,26 @@ def send_snap(self, recipients, snap):
 
         self.log("Sending snap %s to %s" % (snap.snap_id, recipients_str))
 
-        self.client.send(snap.media_id, recipients_str)
+        self.client.send(media_id, recipients_str, snap.duration)
 
     def post_story(self, snap):
-        if not snap.uploaded:
-            self.log("Uploading snap")
-            snap.upload(self)
+        media_id = self._upload_snap(snap)
+        response = self.client.send_to_story(media_id, snap.duration, snap.media_type)
+
+        try:
+            snap.story_id = response['json']['story']['id']
+        except:
+            pass
+
+    def delete_story(self, snap):
+        print snap.story_id 
+        if snap.story_id is None:
+            return
 
-        self.log("Posting snap as story")
-        self.client.send_to_story(snap.media_id, media_type=snap.media_type)
+        self.client._request('delete_story', {
+            'username': self.username,
+            'story_id': snap.story_id
+        })
 
     def add_friend(self, username):
         self.client.add_friend(username)
@@ -111,8 +131,7 @@ def delete_friend(self, username):
     def block(self, username):
         self.client.block(username)
 
-    def get_snaps(self, mark_viewed=True):
-        snaps = self.client.get_snaps()
+    def process_snaps(self, snaps, mark_viewed = True):
         ret = []
 
         for snap_obj in snaps:
@@ -132,3 +151,72 @@ def get_snaps(self, mark_viewed=True):
             ret.append(snap)
 
         return ret
+
+    def process_stories(self, stories):
+        ret = []
+        for snap_obj in stories:
+            media_key = base64.b64decode(snap_obj['media_key'])
+            media_iv = base64.b64decode(snap_obj['media_iv'])
+            data = self.client.get_story_blob(snap_obj['media_id'],
+                                              media_key,
+                                              media_iv)
+            if data is None:
+                continue
+            snap_obj['sender'] = self.username
+            snap = self.process_snap(snap_obj, data, is_story = True)
+            ret.append(snap)
+        return ret
+
+    def get_snaps(self, mark_viewed=True):
+        snaps = self.client.get_snaps()
+        return self.process_snaps(snaps)                
+
+    def get_my_stories(self):
+        response = self.client._request('stories', {
+            'username': self.username
+        })
+        stories = map(lambda s: s['story'], response.json()['my_stories'])
+        return self.process_stories(stories)
+
+    def get_friend_stories(self):
+        response = self.client._request('stories', {
+            'username': self.username
+        })
+        ret = []
+        stories_per_friend = map(lambda s: s['stories'], response.json()['friend_stories'])
+        for stories_obj in stories_per_friend:
+            stories = map(lambda so: so['story'], stories_obj)
+            ret.extend(self.process_stories(stories))
+        return ret
+
+    def clear_stories(self):
+        for story in self.get_my_stories():
+            self.delete_story(story)
+
+    def _upload_snap(self, snap):
+        if not snap.uploaded:
+            snap.media_id = self.client.upload(snap.file.name)
+            snap.uploaded = True
+
+        return snap.media_id
+
+    def _make_request(self, path, data = None, method = 'POST', files = None):
+        if data is None:
+            data = {}
+
+        headers = {
+            'User-Agent': 'Snapchat/8.1.1 (iPhone5,1; iOS 8.1.3; gzip)',
+            'Accept-Language': 'en-US;q=1, en;q=0.9',
+            'Accept-Locale': 'en'
+        }
+
+        now = timestamp()
+
+        if method == 'POST':
+            data['timestamp'] = now
+            data['req_token'] = make_request_token(self.auth_token, str(now))
+            resp = requests.post(BASE_URL + path, data = data, files = files, headers = headers)
+        else:
+            resp = requests.get(BASE_URL + path, params = data, headers = headers)
+
+        return resp

snapchat_bots/constants.py

@@ -1,7 +1,10 @@
+BASE_URL = 'https://feelinsonice-hrd.appspot.com'
+STATIC_TOKEN = 'm198sOkJEn37DjqZ32lpRu76xmw288xSQ9'
+
 DEFAULT_TIMEOUT = 15
 DEFAULT_DURATION = 5
 
-SNAP_IMAGE_DIMENSIONS = (290, 600)
+SNAP_IMAGE_DIMENSIONS = (1334, 750)
 
 MEDIA_TYPE_UNKNOWN = -1
 MEDIA_TYPE_IMAGE = 0

snapchat_bots/exceptions.py

@@ -1,2 +1,5 @@
 class UnknownMediaType(Exception):
     pass
+
+class CannotOpenFile(Exception):
+    pass