--- /dev/null
+diff --git a/bugwarrior/services/mantisbt.py b/bugwarrior/services/mantisbt.py
+new file mode 100644
+index 0000000..e54af0d
+--- /dev/null
++++ b/bugwarrior/services/mantisbt.py
+@@ -0,0 +1,361 @@
++from builtins import filter
++import re
++import six
++
++import requests
++from jinja2 import Template
++
++from bugwarrior.config import asbool, aslist, die
++from bugwarrior.services import IssueService, Issue, ServiceClient
++
++import logging
++log = logging.getLogger(__name__)
++
++
++class MantisbtClient(ServiceClient):
++ def __init__(self, host, token):
++ self.host = host
++ self.session = requests.Session()
++ self.session.headers['Authorization'] = token
++
++ def _api_url(self, path, **context):
++ """ Build the full url to the API endpoint """
++ baseurl = "https://{}/api/rest".format(self.host)
++ return baseurl + path.format(**context)
++
++ def get_user(self):
++ return self.json_response(self.session.get(self._api_url("/users/me")))
++
++ def get_projects(self):
++ return self._getter(self._api_url("/projects"), subkey="projects")
++
++ def get_issues(self):
++ url = self._api_url("/issues?page_size=50")
++ return self._getter(url, page_size=50, subkey="issues")
++
++ def get_assigned_issues(self):
++ """ Returns all issues assigned to authenticated user.
++ """
++ url = self._api_url("/issues?page_size=50&filter_id=assigned")
++ return self._getter(url, page_size=50, subkey="issues")
++
++ def get_monitored_issues(self):
++ """ Returns all issues monitored by authenticated user.
++ """
++ url = self._api_url("/issues?page_size=50&filter_id=monitored")
++ return self._getter(url, page_size=50, subkey="issues")
++
++ def get_reported_issues(self):
++ """ Returns all issues reported by authenticated user.
++ """
++ url = self._api_url("/issues?page_size=50&filter_id=reported")
++ return self._getter(url, page_size=50, subkey="issues")
++
++ def _getter(self, url, page_size=None, subkey=None):
++ """ Pagination utility. Obnoxious. """
++
++ results = []
++ link = dict(next=url)
++ page_number = 1
++
++ while 'next' in link:
++ if page_size is not None:
++ response = self.session.get(link['next'] + "&page=" + str(page_number))
++ else:
++ response = self.session.get(link['next'])
++
++ json_res = self.json_response(response)
++
++ if subkey is not None:
++ json_res = json_res[subkey]
++
++ results += json_res
++
++ if page_size is not None and len(json_res) == page_size:
++ page_number += 1
++ else:
++ break
++
++ return results
++
++class MantisbtIssue(Issue):
++ TITLE = 'mantisbttitle'
++ BODY = 'mantisbtbody'
++ CREATED_AT = 'mantisbtcreatedon'
++ UPDATED_AT = 'mantisbtupdatedat'
++ CLOSED_AT = 'mantisbtclosedon'
++ URL = 'mantisbturl'
++ PROJECT = 'mantisbtproject'
++ NUMBER = 'mantisbtnumber'
++ USER = 'mantisbtuser'
++ CATEGORY = 'mantisbtcategory'
++ STATE = 'mantisbtstate'
++
++ UDAS = {
++ TITLE: {
++ 'type': 'string',
++ 'label': 'Mantisbt Title',
++ },
++ BODY: {
++ 'type': 'string',
++ 'label': 'Mantisbt Body',
++ },
++ CREATED_AT: {
++ 'type': 'date',
++ 'label': 'Mantisbt Created',
++ },
++ UPDATED_AT: {
++ 'type': 'date',
++ 'label': 'Mantisbt Updated',
++ },
++ CLOSED_AT: {
++ 'type': 'date',
++ 'label': 'Mantisbt Closed',
++ },
++ PROJECT: {
++ 'type': 'string',
++ 'label': 'Mantisbt Project',
++ },
++ URL: {
++ 'type': 'string',
++ 'label': 'Mantisbt URL',
++ },
++ NUMBER: {
++ 'type': 'numeric',
++ 'label': 'Mantisbt Issue #',
++ },
++ USER: {
++ 'type': 'string',
++ 'label': 'Mantisbt User',
++ },
++ CATEGORY: {
++ 'type': 'string',
++ 'label': 'Mantisbt Category',
++ },
++ STATE: {
++ 'type': 'string',
++ 'label': 'Mantisbt State',
++ }
++ }
++ UNIQUE_KEY = (URL, NUMBER, )
++
++ def _normalize_tag(self, label):
++ return re.sub(r'[^a-zA-Z0-9]', '_', label)
++
++ def to_taskwarrior(self):
++ body = self.record.get('description')
++ if body:
++ body = body.replace('\r\n', '\n')
++
++ created = self.parse_date(self.record.get('created_at'))
++ updated = self.parse_date(self.record.get('updated_at'))
++ closed_date = None
++ if self.record["status"]["name"] in ["closed", "resolved"]:
++ for history in self.record.get("history", []):
++ if history.get("field", {}).get("name", "") == "status"\
++ and history.get("new_value", {}).get("name", "") in ["closed", "resolved"]:
++ closed_date = history["created_at"]
++ closed = self.parse_date(closed_date)
++
++ return {
++ 'project': self.record['project']['name'],
++ 'priority': self.origin['default_priority'],
++ 'annotations': self.get_annotations(),
++ 'tags': self.get_tags(),
++ 'entry': created,
++ 'end': closed,
++
++ self.TITLE: self.record.get('summary'),
++ self.BODY: body,
++ self.CREATED_AT: created,
++ self.UPDATED_AT: updated,
++ self.CLOSED_AT: closed,
++ self.URL: self.get_url(),
++ self.PROJECT: self.record['project'].get('name'),
++ self.NUMBER: self.record['id'],
++ self.USER: self.record['reporter'].get('name'),
++ self.CATEGORY: self.record['category'].get('name'),
++ self.STATE: self.record['status'].get('label'),
++ }
++
++ def get_url(self):
++ return "https://{}view.php?id={}".format(self.extra['host'], self.record["id"])
++
++ def get_annotations(self):
++ annotations = []
++
++ context = self.record.copy()
++ annotation_template = Template(self.origin['annotation_template'])
++
++ for annotation_dict in self.record.get('notes', []):
++ context.update({
++ 'text': annotation_dict['text'],
++ 'date': annotation_dict['created_at'],
++ 'author': annotation_dict['reporter'].get('name', 'unknown'),
++ 'view': annotation_dict['view_state']['label'],
++ })
++ annotations.append(
++ annotation_template.render(context)
++ )
++ return annotations
++
++ def get_tags(self):
++ tags = []
++
++ context = self.record.copy()
++ tag_template = Template(self.origin['tag_template'])
++
++ for tag_dict in self.record.get('tags', []):
++ context.update({
++ 'tag': self._normalize_tag(tag_dict['name'])
++ })
++ tags.append(
++ tag_template.render(context)
++ )
++
++ return tags
++
++ def get_default_description(self):
++ return self.build_default_description(
++ title=self.record['summary'],
++ url=self.get_processed_url(self.get_url()),
++ number=self.record['id'],
++ )
++
++
++class MantisbtService(IssueService):
++ ISSUE_CLASS = MantisbtIssue
++ CONFIG_PREFIX = 'mantisbt'
++
++ def __init__(self, *args, **kw):
++ super(MantisbtService, self).__init__(*args, **kw)
++
++ self.host = self.config.get('host', 'www.mantisbt.org/bugs/')
++
++ token = self.get_password('token')
++
++ self.client = MantisbtClient(self.host, token)
++ self.user = None
++
++ self.exclude_projects = self.config.get('exclude_projects', [], aslist)
++ self.include_projects = self.config.get('include_projects', [], aslist)
++
++ self.involved_issues = self.config.get(
++ 'involved_issues', default=True, to_type=asbool
++ )
++ self.assigned_issues = self.config.get(
++ 'assigned_issues', default=False, to_type=asbool
++ )
++ self.monitored_issues = self.config.get(
++ 'monitored_issues', default=False, to_type=asbool
++ )
++ self.reported_issues = self.config.get(
++ 'reported_issues', default=False, to_type=asbool
++ )
++ self.tag_template = self.config.get(
++ 'tag_template', default='{{tag}}', to_type=six.text_type
++ )
++ self.annotation_template = self.config.get(
++ 'annotation_template', default='{{date}} {{author}} ({{view}}): {{text}}', to_type=six.text_type
++ )
++
++ def get_service_metadata(self):
++ return {
++ 'tag_template': self.tag_template,
++ 'annotation_template': self.annotation_template,
++ }
++
++ def filter_involved_issues(self, issue):
++ _, issue = issue
++ user = self.client.get_user()
++ uid = user["id"]
++ if issue["reporter"]["id"] != uid and \
++ issue.get("handler", {}).get("id") != uid and \
++ all([ x.get("user", {}).get("id") != uid for x in issue.get("history", [])]) and \
++ all([ x.get("user", {}).get("id") != uid for x in issue.get("monitors", [])]):
++ return False
++ return self.filter_project_name(issue["project"]["name"])
++
++ def filter_issues(self, issue):
++ _, issue = issue
++ return self.filter_project_name(issue["project"]["name"])
++
++ def filter_project_name(self, name):
++ if self.exclude_projects:
++ if name in self.exclude_projects:
++ return False
++
++ if self.include_projects:
++ if name in self.include_projects:
++ return True
++ else:
++ return False
++
++ return True
++
++ @staticmethod
++ def get_keyring_service(service_config):
++ host = service_config.get('host', 'www.mantisbt.org/bugs/')
++ username = service_config.get('username', default='nousername')
++ return "mantisbt://{username}@{host}".format(username=username,
++ host=host)
++
++ @staticmethod
++ def to_issue_dict(issues):
++ return { i['id']: i for i in issues }
++
++ def get_owner(self, issue):
++ return issue.get("handler", {}).get("name")
++
++ def get_author(self, issue):
++ return issue.get("reporter", {}).get("name")
++
++ def issues(self):
++ issues = {}
++ is_limited = self.assigned_issues or self.monitored_issues or self.reported_issues
++
++ if self.assigned_issues:
++ issues.update(
++ filter(self.filter_issues, self.to_issue_dict(self.client.get_assigned_issues()).items())
++ )
++ if self.monitored_issues:
++ issues.update(
++ filter(self.filter_issues, self.to_issue_dict(self.client.get_monitored_issues()).items())
++ )
++ if self.reported_issues:
++ issues.update(
++ filter(self.filter_issues, self.to_issue_dict(self.client.get_reported_issues()).items())
++ )
++
++ if not is_limited:
++ all_issues = self.to_issue_dict(self.client.get_issues())
++ if self.involved_issues:
++ issues.update(
++ filter(self.filter_involved_issues, all_issues.items())
++ )
++ else:
++ issues.update(
++ filter(self.filter_issues, all_issues.items())
++ )
++
++ log.debug(" Found %i issues.", len(issues))
++ if not is_limited:
++ issues = list(filter(self.include, issues.values()))
++ else:
++ issues = list(issues.values())
++ log.debug(" Pruned down to %i issues.", len(issues))
++
++ for issue in issues:
++ issue_obj = self.get_issue_for_record(issue)
++ extra = {
++ 'host': self.host
++ }
++ issue_obj.update_extra(extra)
++ yield issue_obj
++
++ @classmethod
++ def validate_config(cls, service_config, target):
++ if 'token' not in service_config:
++ die("[%s] has no 'mantisbt.token'" % target)
++
++ super(MantisbtService, cls).validate_config(service_config, target)
+diff --git a/setup.py b/setup.py
+index d6d957a..665e36e 100644
+--- a/setup.py
++++ b/setup.py
+@@ -80,6 +80,7 @@ setup(name='bugwarrior',
+ activecollab2=bugwarrior.services.activecollab2:ActiveCollab2Service
+ activecollab=bugwarrior.services.activecollab:ActiveCollabService
+ jira=bugwarrior.services.jira:JiraService
++ mantisbt=bugwarrior.services.mantisbt:MantisbtService
+ megaplan=bugwarrior.services.megaplan:MegaplanService
+ phabricator=bugwarrior.services.phab:PhabricatorService
+ versionone=bugwarrior.services.versionone:VersionOneService
projects / perso / Immae / Config / Nix.git / commitdiff
