FogBugz » FogBugzPy A Python wrapper for the FogBugz API Read More
Clone URL:  
Pushed to one repository · View In Graph Contained in master

Use 'in' rather than deprecated has_key

--HG--
extra : rebase_source : bff356f2d6c75a9fb1fbd4fdc8ba1a1bc1c50cbc

Changeset e34b80087eb7…

Parent 64a97853df52…

by Jason R. Coombs

Changes to one file · Browse files at e34b80087eb7 Showing diff from parent 64a97853df52 Diff from another changeset...

Change 1 of 1 Show Changes Only fogbugz.py Stacked
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
 
162
163
164
165
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
 
161
162
163
164
165
 import sys  try:   from email.generator import _make_boundary  except ImportError:   from mimetools import choose_boundary as _make_boundary  try:   import urllib.request as urllib_request  except ImportError:   import urllib2 as urllib_request  try:   from io import BytesIO  except ImportError:   from StringIO import StringIO as BytesIO    try:   basestring  except NameError:   basestring = str    from bs4 import BeautifulSoup, CData    DEBUG = False # Set to True for debugging output.    class FogBugzAPIError(Exception):   pass    class FogBugzLogonError(FogBugzAPIError):   pass    class FogBugzConnectionError(FogBugzAPIError):   pass    class FogBugz:   def __init__(self, url, token=None):   self.__handlerCache = {}   if not url.endswith('/'):   url += '/'     if token:   self._token = token.encode('utf-8')   else:   self._token = None     self._opener = urllib_request.build_opener()   try:   soup = BeautifulSoup(self._opener.open(url + 'api.xml'))   except (urllib_request.URLError, urllib_request.HTTPError):   e = sys.exc_info()[1]   raise FogBugzConnectionError("Library could not connect to the FogBugz API. Either this installation of FogBugz does not support the API, or the url, %s, is incorrect.\n\nError: %s" % (self._url, e))   self._url = url + soup.response.url.string   self.currentFilter = None     def logon(self, username, password):   """   Logs the user on to FogBugz.     Returns None for a successful login.   """   if self._token:   self.logoff()   try:   response = self.__makerequest('logon', email=username, password=password)   except FogBugzAPIError:   e = sys.exc_info()[1]   raise FogBugzLogonError(e)     self._token = response.token.string   if type(self._token) == CData:   self._token = self._token.encode('utf-8')     def logoff(self):   """   Logs off the current user.   """   self.__makerequest('logoff')   self._token = None     def token(self,token):   """   Set the token without actually logging on. More secure.   """   self._token = token.encode('utf-8')     def __encode_multipart_formdata(self, fields, files):   """   fields is a sequence of (key, value) elements for regular form fields.   files is a sequence of (filename, filehandle) files to be uploaded   returns (content_type, body)   """   BOUNDARY = _make_boundary()     if len(files) > 0:   fields['nFileCount'] = str(len(files))     crlf = '\r\n'   buf = BytesIO()     for k, v in fields.items():   if DEBUG:   print("field: %s: %s"% (repr(k), repr(v)))   buf.write(crlf.join([ '--' + BOUNDARY, 'Content-disposition: form-data; name="%s"' % k, '', v, '' ]).encode('utf-8'))     n = 0   for f, h in files.items():   n += 1   buf.write(crlf.join([ '--' + BOUNDARY, 'Content-disposition: form-data; name="File%d"; filename="%s"' % ( n, f), '' ]).encode('utf-8'))   buf.write(crlf.join([ 'Content-type: application/octet-stream', '', '' ]).encode('utf-8'))   buf.write(h.read().encode('utf-8'))   buf.write(crlf.encode('utf-8'))     buf.write(('--' + BOUNDARY + '--' + crlf).encode('utf-8'))   content_type = "multipart/form-data; boundary=%s" % BOUNDARY   return content_type, buf.getvalue()     def __makerequest(self, cmd, **kwargs):   kwargs["cmd"] = cmd   if self._token:   kwargs["token"] = self._token     fields = kwargs   files = fields.get('Files', {})   if 'Files' in fields:   del fields['Files']     content_type, body = self.__encode_multipart_formdata(fields, files)   if DEBUG:   print(body)   headers = { 'Content-Type': content_type,   'Content-Length': str(len(body))}     try:   url = self._url   if sys.version_info() < (3,):   url = self._url.encode('utf-8')   request = urllib_request.Request(url, body, headers)   response = BeautifulSoup(self._opener.open(request)).response   except urllib_request.URLError:   e = sys.exc_info()[1]   raise FogBugzConnectionError(e)   except UnicodeDecodeError:   e = sys.exc_info()[1]   print(kwargs)   raise     if response.error:   raise FogBugzAPIError('Error Code %s: %s' % (response.error['code'], response.error.string,))   return response     def __getattr__(self, name):   """   Handle all FogBugz API calls.     >>> fb.logon(email@example.com, password)   >>> response = fb.search(q="assignedto:email")   """     # Let's leave the private stuff to Python   if name.startswith("__"):   raise AttributeError("No such attribute '%s'" % name)   - if not self.__handlerCache.has_key(name): + if name not in self.__handlerCache:   def handler(**kwargs):   return self.__makerequest(name, **kwargs)   self.__handlerCache[name] = handler   return self.__handlerCache[name]