__init__.py (5934B)

---

 #Embedded file name: C:\BuildAgent\work\20d73f1b24eed06e\eve\release\V24.01\packages\bugreporter\__init__.py
 from __future__ import print_function
 import requests
 from bugreporter.util import GuessMimeType
 import json
 import mimetools
 import logging
 from cStringIO import StringIO
 import uthread2
 
 class BugReporter(object):
 
     def __init__(self, serverUrl, cacheFolder = None):
         if cacheFolder is None:
             cacheFolder = '.cache'
         self.cacheFolder = cacheFolder
         self.logger = logging.getLogger('bugreporter')
         self.baseUrl = serverUrl
 
     def GetCategories(self):
         ret = []
         for each in self._HttpGet('eve/issue/categories')['categories']:
             ret.append((each['value'], each['id']))
 
         return ret
 
     def SendBugReport(self, categoryName, title, description, reproSteps, sessionInfo, computerInfo, userID, labels, server, build, files):
         data = self._GetData(categoryName, title, description, reproSteps, sessionInfo, computerInfo, userID, labels, server, build)
         response = self._HttpPost('eve/issue/new', data)
         issueID = response['id']
         self._SendAttachments(issueID, files)
         return response
 
     def IsBugsServiceOnline(self):
         try:
             return self._HttpGet('status/health')['isOnline']
         except KeyError:
             return False
         except Exception:
             self.logger.exception('Bugs service not answering IsOnline')
             return False
 
     def _SendAttachments(self, issueID, files):
         try:
             for fileName, fileData in files:
                 self._SendAttachment(issueID, fileName, fileData)
 
         except Exception as e:
             self.logger.exception(e)
             raise SendAttachmentError()
 
     def _SendAttachment(self, issueID, fileName, fileData, retries = 5):
         contentType, buffer = self._MultipartEncode(fileName, fileData)
         urlAddition = 'eve/issue/{}/attachments/add'.format(issueID)
         exception = None
         while retries > 0 or exception is None:
             try:
                 self._Request(urlAddition, 'POST', body=buffer, contentType=contentType)
                 return
             except requests.ConnectionError as exception:
                 self.logger.info(exception)
                 uthread2.Sleep(2.0 / 2 ** retries)
                 retries -= 1
 
         raise exception
 
     def _MultipartEncode(self, fileName, fileData):
         boundary = mimetools.choose_boundary()
         buf = self._GetBufferForFile(boundary, fileData, fileName)
         contentType = self._GetContentTypeForFile(boundary)
         return (contentType, buf)
 
     def _GetContentTypeForFile(self, boundary):
         contentType = 'multipart/form-data; boundary={}'.format(boundary)
         return contentType
 
     def _GetBufferForFile(self, boundary, fileData, fileName):
         buf = StringIO()
         buf.write('--%s\r\n' % boundary)
         buf.write('Content-Disposition: form-data; name="File"; filename="%s"\r\n' % (fileName,))
         buf.write('Content-Type: %s\r\n' % GuessMimeType(fileName))
         buf.write('\r\n' + fileData + '\r\n')
         buf.write('--{}--\r\n'.format(boundary))
         buf = buf.getvalue()
         return buf
 
     def _GetData(self, categoryID, title, description, reproSteps, sessionInfo, computerInfo, userID, labels, server, build):
         description = '{} \n\nSession Info:\n{}'.format(description, sessionInfo)
         return {'categoryID': str(categoryID),
          'summary': title,
          'description': description,
          'reproductionSteps': reproSteps,
          'computerInfo': computerInfo,
          'originalReporterId': '{}'.format(userID),
          'Server': server,
          'Build': build,
          'labels': labels}
 
     def _HttpGet(self, urlAddition):
         return json.loads(self._Request(urlAddition, 'GET'))
 
     def _HttpPost(self, urlAddition, data):
         return json.loads(self._Request(urlAddition, 'POST', json.dumps(data)))
 
     def _Request(self, urlAddition, method, body = None, contentType = None):
         self.logger.info("making a '%s' call to the bugs service '%s'", method, urlAddition)
         url = '{}/{}'.format(self.baseUrl, urlAddition)
         self.logger.info('url: {}'.format(url))
         if contentType is None:
             contentType = 'application/json'
         headers = {}
         token = sm.GetService('connection').GetAccessToken()
         if token is not None:
             headers['Authorization'] = 'Bearer {}'.format(token)
         if contentType is not None:
             headers['Content-Type'] = contentType
         if method == 'GET':
             resp = requests.get(url, headers=headers)
         elif method == 'POST':
             resp = requests.post(url, data=body, headers=headers)
         else:
             raise RuntimeError('BugService unsupported method %s' % method)
         self.logger.info('response: {}'.format(resp))
         content = resp.content
         self.CheckResponse(resp, method)
         return content
 
     def CheckResponse(self, resp, method):
         status_code = resp.status_code
         if method == 'POST' and status_code not in (200, 201):
             self.logger.error('Bug Service Error Code: %s', status_code, extra={'response': resp,
              'fingerprint': ['bugservice', status_code]})
             raise RuntimeError('Unexpected error from the bugs service')
 
 
 class SendAttachmentError(Exception):
     pass
 
 
 def SendTestBugReport(serverUrl):
     bugReporter = BugReporter(serverUrl, '.cache')
     categoryName, categoryID = bugReporter.GetCategories()[0]
     response = bugReporter.SendBugReport(categoryID, 'title', 'description', 'repro steps', 'sessionInfo', 'computerInfo', 'charID', ['UI'], [])
     print("Bug Report successfully created '{}'".format(response['self']))
 
 
 if __name__ == '__main__':
     import sys
     if len(sys.argv) > 1:
         SendTestBugReport(sys.argv[1])