Initial import

Author: mpolden

.gitignore

@@ -0,0 +1,2 @@
+*.pyc
+.vagrant/

LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2013 Martin Polden
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

Makefile

@@ -0,0 +1,7 @@
+all: lint test
+
+test:
+	python pysnap/tests.py
+
+lint:
+	flake8 --max-complexity=10 pysnap/*.py

README.md

@@ -0,0 +1,9 @@
+pysnap
+======
+Implementation of Snapchat API in Python.
+
+Dependencies
+------------
+
+    pip install --use-mirrors -r requirements.txt
+

Vagrantfile

@@ -0,0 +1,31 @@
+# -*- mode: ruby -*-
+# vi: set ft=ruby :
+
+$script = <<SCRIPT
+  set -x
+
+  # Ensure noninteractive apt-get
+  export DEBIAN_FRONTEND=noninteractive
+
+  # Set time zone
+  echo "Europe/Oslo" > /etc/timezone
+  dpkg-reconfigure tzdata
+
+  # Install packages
+  apt-get -y --quiet update
+  apt-get -y --quiet install make python-pip
+
+  # Install pip packages
+  pip install --quiet --upgrade -r /vagrant/requirements.txt \
+          -r /vagrant/dev-requirements.txt
+SCRIPT
+
+Vagrant.configure("2") do |config|
+  config.vm.box = "raring64-current"
+  config.vm.box_url = "http://cloud-images.ubuntu.com/vagrant/raring/current/raring-server-cloudimg-amd64-vagrant-disk1.box"
+  config.ssh.forward_agent = true
+  config.vm.provider :virtualbox do |vb|
+    vb.customize ["modifyvm", :id, "--memory", "1024"]
+  end
+  config.vm.provision :shell, :inline => $script
+end

dev-requirements.txt

@@ -0,0 +1,7 @@
+bpython==0.12
+flake8==2.0
+mccabe==0.2.1
+pep8==1.4.6
+pip-tools==0.3.4
+pyflakes==0.7.3
+Pygments==1.6

pysnap/pysnap.py

@@ -0,0 +1,92 @@
+#!/usr/bin/env python
+
+import requests
+from Crypto.Cipher import AES
+from hashlib import sha256
+from time import time
+from urlparse import urljoin
+
+URL = 'https://feelinsonice-hrd.appspot.com/bq/'
+SECRET = 'iEk21fuwZApXlz93750dmW22pw389dPwOk'
+STATIC_TOKEN = 'm198sOkJEn37DjqZ32lpRu76xmw288xSQ9'
+BLOB_ENCRYPTION_KEY = 'M02cnQ51Ji97vwT4'
+HASH_PATTERN = ('00011101111011100011110101011110'
+                '11010001001110011000110001000110')
+
+
+def make_request_token(a, b):
+    hash_a = sha256(SECRET + a).hexdigest()
+    hash_b = sha256(b + SECRET).hexdigest()
+    result = [None] * len(HASH_PATTERN)
+    for i, c in enumerate(HASH_PATTERN):
+        result[i] = hash_b[i] if c == '1' else hash_a[i]
+    return ''.join(result)
+
+
+def pkcs5_pad(data, blocksize=16):
+    pad_count = blocksize - len(data) % blocksize
+    return data + (chr(pad_count) * pad_count)
+
+
+def is_video(data):
+    return len(data) > 1 and data[0] == chr(0x00) and data[1] == chr(0x00)
+
+
+def is_image(data):
+    return len(data) > 1 and data[0] == chr(0xFF) and data[1] == chr(0xD8)
+
+
+def decrypt(data):
+    cipher = AES.new(BLOB_ENCRYPTION_KEY, AES.MODE_ECB)
+    return cipher.decrypt(pkcs5_pad(data))
+
+
+def encrypt(data):
+    cipher = AES.new(BLOB_ENCRYPTION_KEY, AES.MODE_ECB)
+    return cipher.encrypt(pkcs5_pad(data))
+
+
+def timestamp():
+    return int(round(time() * 1000))
+
+
+class Snapchat(object):
+
+    def __init__(self, username, password):
+        self.username = username
+        self.password = password
+
+    def _request(self, endpoint, data=None):
+        now = timestamp()
+        if data is None:
+            data = {}
+        data.update({
+            'username': self.username,
+            'timestamp': now,
+            'req_token': make_request_token(
+                getattr(self, 'auth_token', STATIC_TOKEN), str(now))
+        })
+        return requests.post(urljoin(URL, endpoint), data=data)
+
+    def login(self):
+        r = self._request('login', {'password': self.password})
+        result = r.json()
+        if 'auth_token' in result:
+            self.auth_token = result['auth_token']
+        return result
+
+    def logout(self):
+        r = self._request('logout')
+        return r.status_code == 200
+
+    def get_updates(self, update_timestamp=0):
+        r = self._request('updates', {'update_timestamp': update_timestamp})
+        result = r.json()
+        if 'auth_token' in result:
+            self.auth_token = result['auth_token']
+        return result
+
+    def get_blob(self, snap_id):
+        r = self._request('blob', {'id': snap_id})
+        data = decrypt(r.content)
+        return data if is_image(data) or is_video(data) else None

pysnap/tests.py

@@ -0,0 +1,31 @@
+#!/usr/bin/env python
+
+import unittest
+from pysnap import make_request_token, is_image, is_video, pkcs5_pad
+
+
+class Snapchat(unittest.TestCase):
+
+    def test_make_request_token(self):
+        self.assertEqual(('c6a633f0c2c9e72f7cdf49f888fc64ee7179096c57fdedaad1f'
+                          '4ea513dbb7b34'), make_request_token('foo', 'bar'))
+
+    def test_is_image(self):
+        self.assertFalse(is_image([]))
+        self.assertFalse(is_image([None]))
+        self.assertFalse(is_image([None, None]))
+        self.assertTrue(is_image([chr(0xFF), chr(0xD8)]))
+
+    def test_is_video(self):
+        self.assertFalse(is_video([]))
+        self.assertFalse(is_video([None]))
+        self.assertFalse(is_video([None, None]))
+        self.assertTrue(is_video([chr(0x00), chr(0x00)]))
+
+    def test_pkcs5_pad(self):
+        self.assertEqual('\x10' * 16, pkcs5_pad(''))
+        self.assertEqual('foo\r\r\r\r\r\r\r\r\r\r\r\r\r', pkcs5_pad('foo'))
+
+
+if __name__ == '__main__':
+    unittest.main()

requirements.txt

@@ -0,0 +1,2 @@
+pycrypto==2.6
+requests==1.2.3