1 from __future__ import absolute_import
2 __copyright__ = "Copyright (C) 2013 David Braam - Released under terms of the AGPLv3 License"
5 import httplib as httpclient
8 class httpUploadDataStream(object):
9 def __init__(self, progressCallback):
13 self._progressCallback = progressCallback
15 def write(self, data):
20 for n in xrange(0, blocks):
21 self._dataList.append(data[n*2048:n*2048+2048])
22 self._dataList.append(data[blocks*2048:])
23 self._totalLength += size
26 if self._readPos >= len(self._dataList):
28 ret = self._dataList[self._readPos]
30 if self._progressCallback is not None:
31 self._progressCallback(float(self._readPos / len(self._dataList)))
35 return self._totalLength
37 class Youmagine(object):
38 def __init__(self, authToken, progressCallback = None):
39 self._hostUrl = 'api.youmagine.com'
40 self._viewUrl = 'www.youmagine.com'
41 self._authUrl = 'https://www.youmagine.com/integrations/cura/authorized_integrations/new'
42 self._authToken = authToken
46 self._hostReachable = True
47 self._progressCallback = progressCallback
58 ('3D printer parts and enhancements', 11),
62 ('Creative Commons - Attribution Share Alike', 'ccbysa'),
63 ('Creative Commons - Attribution Non-Commercial ShareAlike', 'ccbyncsa'),
64 ('Creative Commons - Attribution No Derivatives', 'ccbynd'),
65 ('Creative Commons - Attribution Non-Commercial No Derivatives', 'ccbyncsa'),
69 def getAuthorizationUrl(self):
72 def getCategories(self):
73 return map(lambda n: n[0], self._categories)
75 def getLicenses(self):
76 return map(lambda n: n[0], self._licenses)
78 def setAuthToken(self, token):
79 self._authToken = token
83 def getAuthToken(self):
84 return self._authToken
86 def isHostReachable(self):
87 return self._hostReachable
89 def viewUrlForDesign(self, id):
90 return 'https://%s/designs/%d' % (self._viewUrl, id)
92 def editUrlForDesign(self, id):
93 return 'https://%s/designs/%d/edit' % (self._viewUrl, id)
95 def isAuthorized(self):
96 if self._authToken is None:
98 if self._userName is None:
99 #No username yet, try to request the username to see if the authToken is valid.
100 result = self._request('GET', '/authorized_integrations/%s/whoami.json' % (self._authToken))
102 if 'error' in result:
103 self._authToken = None
105 self._userName = result['screen_name']
106 self._userID = result['id']
109 def createDesign(self, name, description, category, license):
110 res = self._request('POST', '/designs.json', {'design[name]': name, 'design[excerpt]': description, 'design[design_category_id]': filter(lambda n: n[0] == category, self._categories)[0][1], 'design[license]': filter(lambda n: n[0] == license, self._licenses)[0][1]})
116 def publishDesign(self, id):
117 res = self._request('PUT', '/designs/%d/mark_as/publish.json' % (id), {'ignore': 'me'})
122 def createDocument(self, designId, name, contents):
123 res = self._request('POST', '/designs/%d/documents.json' % (designId), {'document[name]': name, 'document[description]': 'Uploaded from Cura'}, {'document[file]': (name, contents)})
129 def createImage(self, designId, name, contents):
130 res = self._request('POST', '/designs/%d/images.json' % (designId), {'image[name]': name, 'image[description]': 'Uploaded from Cura'}, {'image[file]': (name, contents)})
136 def listDesigns(self):
137 res = self._request('GET', '/users/%s/designs.json' % (self._userID))
140 def _request(self, method, url, postData = None, files = None):
142 if self._authToken is not None:
143 url += '?auth_token=%s' % (self._authToken)
144 error = 'Failed to connect to %s' % self._hostUrl
145 for n in xrange(0, retryCount):
146 if self._http is None:
147 self._http = httpclient.HTTPSConnection(self._hostUrl)
149 if files is not None:
150 boundary = 'wL36Yn8afVp8Ag7AmP8qZ0SA4n1v9T'
151 s = httpUploadDataStream(self._progressCallback)
152 for k, v in files.iteritems():
155 s.write('--%s\r\n' % (boundary))
156 s.write('Content-Disposition: form-data; name="%s"; filename="%s"\r\n' % (k, filename))
157 s.write('Content-Type: application/octet-stream\r\n')
158 s.write('Content-Transfer-Encoding: binary\r\n')
160 s.write(fileContents)
163 for k, v in postData.iteritems():
164 s.write('--%s\r\n' % (boundary))
165 s.write('Content-Disposition: form-data; name="%s"\r\n' % (k))
169 s.write('--%s--\r\n' % (boundary))
171 self._http.request(method, url, s, {"Content-type": "multipart/form-data; boundary=%s" % (boundary), "Content-Length": len(s)})
172 elif postData is not None:
173 self._http.request(method, url, urllib.urlencode(postData), {"Content-type": "application/x-www-form-urlencoded"})
175 self._http.request(method, url)
180 response = self._http.getresponse()
181 responseText = response.read()
186 if responseText == '':
188 return json.loads(responseText)
190 print response.getheaders()
192 error = 'Failed to decode JSON response'
193 self._hostReachable = False
194 return {'error': error}
197 #Fake Youmagine class to test without internet
198 class FakeYoumagine(Youmagine):
199 def __init__(self, authToken, callback):
200 super(FakeYoumagine, self).__init__(authToken)
201 self._authUrl = 'file:///C:/Models/output.html'
202 self._authToken = None
204 def isAuthorized(self):
205 if self._authToken is None:
207 if self._userName is None:
208 self._userName = 'FakeYoumagine'
212 def isHostReachable(self):
215 def createDesign(self, name, description, category, license):
218 def publishDesign(self, id):
221 def createDocument(self, designId, name, contents):
222 print "Create document: %s" % (name)
223 f = open("C:/models/%s" % (name), "wb")
228 def createImage(self, designId, name, contents):
229 print "Create image: %s" % (name)
230 f = open("C:/models/%s" % (name), "wb")
235 def listDesigns(self):
238 def _request(self, method, url, postData = None, files = None):
239 print "Err: Tried to do request: %s %s" % (method, url)