First commit

Author: ingmferrer

salesforce/__init__.py

@@ -0,0 +1,158 @@
+import requests
+from salesforce.decorators import access_token_required
+from salesforce.exceptions import UnknownError, BadOAuthTokenError
+from urllib.parse import unquote, urlencode
+
+
+class Client(object):
+    BASE_URL = '{}services/data/'
+    SALESFORCE_REQUEST_TOKEN_URL = 'https://login.salesforce.com/services/oauth2/token'
+    SALESFORCE_AUTHORIZE_URL = 'https://login.salesforce.com/services/oauth2/authorize'
+
+    def __init__(self, client_id, client_secret, instance_url, version):
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self.instance_url = self.BASE_URL.format(instance_url)
+        self.rest_url = self.instance_url + '{}/'.format(version)
+        if version.startswith('v'):
+            version = version[1:]
+        self.version = version
+        self.access_token = None
+        self.resource_urls = {}
+
+    def _get_resource_url(self, name):
+        url = self.resource_urls.get(name, None)
+        if url:
+            return url
+        user_info = self.get_user_info()
+        for k, v in user_info['urls'].items():
+            self.resource_urls[k] = v.replace('{version}', self.version)
+        return self.resource_urls.get(name, None)
+
+    def set_access_token(self, token):
+        self.access_token = token['access_token']
+
+    def authorization_url(self, redirect_uri):
+        params = {
+            'response_type': 'code',
+            'client_id': self.client_id,
+            'redirect_uri': redirect_uri
+        }
+        url = self.SALESFORCE_AUTHORIZE_URL
+        return '{}?{}'.format(url, urlencode(params))
+
+    def exchange_code(self, redirect_uri, code):
+        headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+        data = {
+            'grant_type': 'authorization_code',
+            'redirect_uri': redirect_uri,
+            'code': unquote(code),
+            'client_id': self.client_id,
+            'client_secret': self.client_secret
+        }
+        return self._request('POST', self.SALESFORCE_REQUEST_TOKEN_URL, data=data, headers=headers)
+
+    def extend_token(self, refresh_token):
+        headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+        data = {
+            'grant_type': 'refresh_token',
+            'refresh_token': unquote(refresh_token),
+            'client_id': self.client_id,
+            'client_secret': self.client_secret
+        }
+        return self._request('POST', self.SALESFORCE_REQUEST_TOKEN_URL, data=data, headers=headers)
+
+    @access_token_required
+    def get_user_info(self):
+        return self._request('GET', 'https://login.salesforce.com/services/oauth2/userinfo')
+
+    def get_versions(self):
+        return self._get(self.instance_url)
+
+    @access_token_required
+    def get_resources_by_version(self):
+        return self._get(self.rest_url)
+
+    @access_token_required
+    def get_limits(self):
+        return self._get(self.rest_url + 'limits/')
+
+    @access_token_required
+    def get_describe_global(self):
+        return self._get(self.rest_url + 'sobjects/')
+
+    @access_token_required
+    def get_sobject(self, sobject):
+        return self._get(self.rest_url + 'sobjects/{}/'.format(sobject))
+
+    @access_token_required
+    def create_sobject(self, sobject, data):
+        return self._post(self.rest_url + 'sobjects/{}/'.format(sobject), json=data)
+
+    @access_token_required
+    def get_sobject_describe(self, sobject):
+        return self._get(self.rest_url + 'sobjects/{}/describe/'.format(sobject))
+
+    def create_apex_class(self, name, body):
+        data = {
+            'ApiVersion': self.version,
+            'Body': body,
+            'Name': name
+        }
+        url = self.rest_url + 'tooling/sobjects/ApexClass'
+        return self._request('POST', url, json=data)
+
+    def create_remote_site(self, name, url):
+        data = '<env:Envelope xmlns:env="http://schemas.xmlsoap.org/soap/envelope/" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"><env:Header><urn:SessionHeader xmlns:urn="http://soap.sforce.com/2006/04/metadata"><urn:sessionId>{sessionId}</urn:sessionId></urn:SessionHeader></env:Header><env:Body><createMetadata xmlns="http://soap.sforce.com/2006/04/metadata"><metadata xsi:type="RemoteSiteSetting"><fullName>{name}</fullName><isActive>true</isActive><url>{url}</url></metadata></createMetadata></env:Body></env:Envelope>'
+        data = data.replace('{name}', name).replace('{url}', url).replace('{sessionId}', self.access_token)
+
+        headers = {
+            'SOAPAction': 'RemoteSiteSetting',
+            'Content-Type': 'text/xml'
+        }
+        url = self._get_resource_url('metadata')
+        return self._request('POST', url, data=data, headers=headers)
+
+    def create_apex_trigger(self, name, body, sobject):
+        data = {
+            'ApiVersion': self.version,
+            'Body': body,
+            'Name': name,
+            'TableEnumOrId': sobject
+        }
+        url = self.rest_url + 'tooling/sobjects/ApexTrigger'
+        return self._request('POST', url, json=data)
+
+    def _get(self, url, **kwargs):
+        return self._request('GET', url, **kwargs)
+
+    def _post(self, url, **kwargs):
+        return self._request('POST', url, **kwargs)
+
+    def _put(self, url, **kwargs):
+        return self._request('PUT', url, **kwargs)
+
+    def _delete(self, url, **kwargs):
+        return self._request('DELETE', url, **kwargs)
+
+    def _request(self, method, url, headers=None, **kwargs):
+        _headers = {
+            'Authorization': 'Bearer {}'.format(self.access_token),
+            'Accept': 'application/json',
+            'Content-Type': 'application/json'
+        }
+        if headers:
+            _headers.update(headers)
+        return self._parse(requests.request(method, url, headers=_headers, **kwargs))
+
+    def _parse(self, response):
+        print(response.status_code)
+        print(response.text)
+        status_code = response.status_code
+        if status_code == 200 or status_code == 201:
+            return response.json()
+        if status_code == 204:
+            return None
+        if status_code == 403:
+            raise BadOAuthTokenError()
+        raise UnknownError()

salesforce/client.py

@@ -0,0 +1,13 @@
+from salesforce.exceptions import AccessTokenRequired
+from functools import wraps
+
+
+def access_token_required(func):
+    @wraps(func)
+    def helper(*args, **kwargs):
+        client = args[0]
+        if not client.access_token:
+            raise AccessTokenRequired('You must set the Access Token.')
+        return func(*args, **kwargs)
+
+    return helper

salesforce/decorators.py

@@ -0,0 +1,14 @@
+class BaseError(Exception):
+    pass
+
+
+class UnknownError(BaseError):
+    pass
+
+
+class AccessTokenRequired(BaseError):
+    pass
+
+
+class BadOAuthTokenError(BaseError):
+    pass

salesforce/exceptions.py

@@ -0,0 +1,11 @@
+from setuptools import setup
+
+setup(name='salesforce',
+      version='0.1',
+      description='API wrapper for Salesforce written in Python',
+      url='https://github.com/GearPlug/salesforce-python',
+      author='Miguel Ferrer',
+      author_email='ingferrermiguel@gmail.com',
+      license='GPL',
+      packages=['salesforce'],
+      zip_safe=False)