Break TaskConstructor out of base Issue class.

bugwarrior/collect.py

@@ -1,9 +1,13 @@
+import copy
 import logging
 import multiprocessing
 import time
 
+from jinja2 import Template
 from pkg_resources import iter_entry_points
 
+from taskw.task import Task
+
 log = logging.getLogger(__name__)
 
 # Sentinels for process completion status
@@ -88,7 +92,7 @@ def aggregate_issues(conf, main_section, debug):
     while currently_running > 0:
         issue = queue.get(True)
         try:
-            yield issue.get_taskwarrior_record()
+            yield TaskConstructor(issue).get_taskwarrior_record()
         except AttributeError:
             if isinstance(issue, tuple):
                 currently_running -= 1
@@ -100,3 +104,50 @@ def aggregate_issues(conf, main_section, debug):
             yield issue
 
     log.info("Done aggregating remote issues.")
+
+
+class TaskConstructor:
+    """ Construct a taskwarrior task from a foreign record. """
+
+    def __init__(self, issue):
+        self.issue = issue
+
+    def get_added_tags(self):
+        added_tags = []
+        for tag in self.issue.config.add_tags:
+            tag = Template(tag).render(self.get_template_context())
+            if tag:
+                added_tags.append(tag)
+
+        return added_tags
+
+    def get_taskwarrior_record(self, refined=True) -> dict:
+        if not getattr(self, '_taskwarrior_record', None):
+            self._taskwarrior_record = self.issue.to_taskwarrior()
+        record = copy.deepcopy(self._taskwarrior_record)
+        if refined:
+            record = self.refine_record(record)
+        if 'tags' not in record:
+            record['tags'] = []
+        if refined:
+            record['tags'].extend(self.get_added_tags())
+        return record
+
+    def get_template_context(self):
+        context = (
+            self.get_taskwarrior_record(refined=False).copy()
+        )
+        context.update(self.issue.extra)
+        context.update({
+            'description': self.issue.get_default_description(),
+        })
+        return context
+
+    def refine_record(self, record):
+        for field in Task.FIELDS.keys():
+            if field in self.issue.config.templates:
+                template = Template(self.issue.config.templates[field])
+                record[field] = template.render(self.get_template_context())
+            elif hasattr(self.issue, 'get_default_%s' % field):
+                record[field] = getattr(self.issue, 'get_default_%s' % field)()
+        return record

bugwarrior/docs/contributing/new-service.rst

@@ -264,7 +264,7 @@ Create a test file and implement at least the minimal service tests by inheritin
 
           expected = { ... }
 
-          self.assertEqual(issue.get_taskwarrior_record(), expected)
+          self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)
 
 9. Documentation
 ------------------

bugwarrior/services/__init__.py

@@ -1,14 +1,11 @@
 import abc
-import copy
 import re
 
 from dateutil.parser import parse as parse_date
 from dateutil.tz import tzlocal
 from jinja2 import Template
 import pytz
 
-from taskw.task import Task
-
 from bugwarrior.config import schema, secrets
 from bugwarrior.db import MARKUP, URLShortener
 
@@ -202,27 +199,6 @@ def get_tags_from_labels(self,
 
         return tags
 
-    def get_added_tags(self):
-        added_tags = []
-        for tag in self.config.add_tags:
-            tag = Template(tag).render(self.get_template_context())
-            if tag:
-                added_tags.append(tag)
-
-        return added_tags
-
-    def get_taskwarrior_record(self, refined=True) -> dict:
-        if not getattr(self, '_taskwarrior_record', None):
-            self._taskwarrior_record = self.to_taskwarrior()
-        record = copy.deepcopy(self._taskwarrior_record)
-        if refined:
-            record = self.refine_record(record)
-        if 'tags' not in record:
-            record['tags'] = []
-        if refined:
-            record['tags'].extend(self.get_added_tags())
-        return record
-
     def get_priority(self):
         return self.PRIORITY_MAP.get(
             self.record.get('priority'),
@@ -272,25 +248,6 @@ def build_default_description(
             url,
         )
 
-    def get_template_context(self):
-        context = (
-            self.get_taskwarrior_record(refined=False).copy()
-        )
-        context.update(self.extra)
-        context.update({
-            'description': self.get_default_description(),
-        })
-        return context
-
-    def refine_record(self, record):
-        for field in Task.FIELDS.keys():
-            if field in self.config.templates:
-                template = Template(self.config.templates[field])
-                record[field] = template.render(self.get_template_context())
-            elif hasattr(self, 'get_default_%s' % field):
-                record[field] = getattr(self, 'get_default_%s' % field)()
-        return record
-
     @property
     def record(self):
         return self._foreign_record
@@ -299,15 +256,6 @@ def record(self):
     def extra(self):
         return self._extra
 
-    def __str__(self):
-        return '%s: %s' % (
-            self.config.target,
-            self.get_taskwarrior_record()['description']
-        )
-
-    def __repr__(self):
-        return '<%s>' % str(self)
-
 
 class ServiceClient:
     """ Abstract class responsible for making requests to service API's. """

tests/test_activecollab.py

@@ -5,6 +5,7 @@
 import pypandoc
 import pytz
 
+from bugwarrior.collect import TaskConstructor
 from bugwarrior.services.activecollab import (
     ActiveCollabService
 )
@@ -141,4 +142,4 @@ def test_issues(self):
             'project': 'something',
             'tags': []}
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)

tests/test_activecollab2.py

@@ -4,6 +4,7 @@
 import pytz
 import responses
 
+from bugwarrior.collect import TaskConstructor
 from bugwarrior.services.activecollab2 import ActiveCollab2Service
 
 from .base import ServiceTest, AbstractServiceTest
@@ -97,4 +98,4 @@ def test_issues(self):
             'project': 'something',
             'tags': []}
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)

tests/test_azuredevops.py

@@ -3,6 +3,7 @@
 
 from dateutil.tz.tz import tzutc
 
+from bugwarrior.collect import TaskConstructor
 from bugwarrior.services.azuredevops import (
     AzureDevopsService,
     striphtml,
@@ -230,7 +231,7 @@ def test_issues(self):
             "tags": []
         }
         issue = next(self.service.issues())
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)
 
     def test_issues_wiql_filter(self):
         expected = {
@@ -255,4 +256,4 @@ def test_issues_wiql_filter(self):
         }
         service = self.get_service(config_overrides={'wiql_filter': 'something'})
         issue = next(service.issues())
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)

tests/test_bitbucket.py

@@ -1,5 +1,6 @@
 import responses
 
+from bugwarrior.collect import TaskConstructor
 from bugwarrior.services.bitbucket import BitbucketService
 
 from .base import ServiceTest, AbstractServiceTest
@@ -104,7 +105,7 @@ def test_issues(self):
             'project': 'somerepo',
             'tags': []}
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected_issue)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected_issue)
 
         expected_pr = {
             'annotations': ['@nobody - Some comment.'],
@@ -116,7 +117,7 @@ def test_issues(self):
             'project': 'somerepo',
             'tags': []}
 
-        self.assertEqual(pr.get_taskwarrior_record(), expected_pr)
+        self.assertEqual(TaskConstructor(pr).get_taskwarrior_record(), expected_pr)
 
     def test_get_owner(self):
         issue = {

tests/test_bts.py

@@ -1,6 +1,8 @@
 import sys
 from unittest import mock, SkipTest
 
+from bugwarrior.collect import TaskConstructor
+
 if sys.version_info >= (3, 11):
     raise SkipTest(
         "Python-3.11+ not supported. "
@@ -89,4 +91,4 @@ def test_issues(self):
             'btsstatus': 'pending',
             'tags': []}
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)

tests/test_bugzilla.py

@@ -2,6 +2,7 @@
 from unittest import mock
 from collections import namedtuple
 
+from bugwarrior.collect import TaskConstructor
 from bugwarrior.services.bz import BugzillaService
 
 from .base import ConfigTest, ServiceTest, AbstractServiceTest
@@ -155,7 +156,7 @@ def test_issues(self):
             'project': 'Something',
             'tags': []}
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)
 
     def test_only_if_assigned(self):
         with mock.patch('bugzilla.Bugzilla'):
@@ -206,7 +207,7 @@ def test_only_if_assigned(self):
             'project': 'Something',
             'tags': []}
 
-        self.assertEqual(next(issues).get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(next(issues)).get_taskwarrior_record(), expected)
 
         # Only one issue is assigned.
         self.assertRaises(StopIteration, lambda: next(issues))
@@ -246,9 +247,9 @@ def test_also_unassigned(self):
 
         issues = self.service.issues()
 
-        self.assertIn(next(issues).get_taskwarrior_record()['bugzillabugid'],
+        self.assertIn(TaskConstructor(next(issues)).get_taskwarrior_record()['bugzillabugid'],
                       [1234567, 1234568])
-        self.assertIn(next(issues).get_taskwarrior_record()['bugzillabugid'],
+        self.assertIn(TaskConstructor(next(issues)).get_taskwarrior_record()['bugzillabugid'],
                       [1234567, 1234568])
         # Only two issues are assigned to the user or unassigned.
         self.assertRaises(StopIteration, lambda: next(issues))

tests/test_deck.py

@@ -4,7 +4,9 @@
 import pydantic
 from dateutil.tz import tzutc
 
+from bugwarrior.collect import TaskConstructor
 from bugwarrior.services.deck import NextcloudDeckClient, NextcloudDeckService
+
 from .base import AbstractServiceTest, ServiceTest
 
 
@@ -157,7 +159,7 @@ def test_issues(self):
             'tags': ['Later']
         }
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)
 
     def test_filter_boards_include(self):
         self.config['deck']['include_board_ids'] = '5'

tests/test_gerrit.py

@@ -2,7 +2,9 @@
 
 import responses
 
+from bugwarrior.collect import TaskConstructor
 from bugwarrior.services.gerrit import GerritService
+
 from .base import ServiceTest, AbstractServiceTest
 
 
@@ -82,7 +84,7 @@ def test_work_in_progress(self):
             'project': 'nova',
             'tags': []}
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)
 
     @responses.activate
     def test_issues(self):
@@ -107,4 +109,4 @@ def test_issues(self):
             'project': 'nova',
             'tags': []}
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)

tests/test_gitbug.py

@@ -4,6 +4,7 @@
 import dateutil
 import pydantic
 
+from bugwarrior.collect import TaskConstructor
 from bugwarrior.services.gitbug import GitBugClient, GitBugConfig, GitBugService
 
 from .base import AbstractServiceTest, ConfigTest, ServiceTest
@@ -82,7 +83,7 @@ def test_issues(self):
             'tags': []
         }
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)
 
 
 class TestGitBugConfig(ConfigTest):

tests/test_github.py

@@ -4,6 +4,7 @@
 import pytz
 import responses
 
+from bugwarrior.collect import TaskConstructor
 from bugwarrior.services.github import (
     GithubConfig, GithubService, GithubClient)
 
@@ -85,7 +86,7 @@ def test_draft(self):
             'tags': []
         }
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)
 
     def test_to_taskwarrior(self):
         service = self.get_mock_service(GithubService, config_overrides={
@@ -178,7 +179,7 @@ def test_issues(self):
             'project': 'arbitrary_repo',
             'tags': []}
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)
 
 
 class TestGithubIssueQuery(AbstractServiceTest, ServiceTest):
@@ -238,7 +239,7 @@ def test_issues(self):
             'project': 'arbitrary_repo',
             'tags': []}
 
-        self.assertEqual(issue.get_taskwarrior_record(), expected)
+        self.assertEqual(TaskConstructor(issue).get_taskwarrior_record(), expected)
 
 
 class TestGithubService(ServiceTest):