PYAPI-30 Portfolio: API Cleanup

Author: astromatt

PKG-INFO

@@ -13,62 +13,79 @@ def __init__(self, plan_id, *args, **kwargs):
 
     def get_epic(self, epic):
         key = [x.get('link', None) for x in epic.get('links', [])]
-        estimates = self.estimates_dict(epic['estimates'])
+        estimates = self.get_estimates_dict(epic['estimates'])
         estimates.update(Total=sum(estimates.values()))
+        team_id = epic.get('teamId', None)
         return {
             'title': epic.get('title', None),
-            'team': self.team_name(epic.get('teamId')) if epic.get('teamId', None) else None,
+            'team': self.get_team_name(team_id) if team_id else None,
             'description': epic.get('description', None),
             'issuekey': key[0] if key else None,
             'estimates': estimates}
 
-    def plan(self):
+    def get_plan(self):
         url = '/rest/roadmap/1.0/plans/{0}.json'.format(self.plan_id)
         return self.get(url)
 
-    def stages(self):
+    def get_stages(self):
         url = '/rest/roadmap/1.0/plans/{0}/stages.json'.format(self.plan_id)
         return self.get(url)
 
-    def teams(self):
+    def get_teams(self):
         url = '/rest/roadmap/1.0/plans/{0}/teams.json'.format(self.plan_id)
         return self.get(url)
 
-    def team_name(self, team_id):
-        return [team['title'] for team in self.teams()['collection'] if team['id'] == str(team_id)][0]
+    def get_team_name(self, team_id):
+        all_teams = self.get_teams()['collection']
+        return [team['title'] for team in all_teams if team['id'] == str(team_id)][0]
 
-    def config(self):
+    def get_config(self):
         url = '/rest/roadmap/1.0/plans/{0}/config.json'.format(self.plan_id)
         return self.get(url)
 
-    def persons(self):
+    def get_persons(self):
         url = '/rest/roadmap/1.0/plans/{0}/persons.json'.format(self.plan_id)
         return self.get(url)
 
-    def streams(self):
+    def get_streams(self):
         url = '/rest/roadmap/1.0/plans/{0}/streams.json'.format(self.plan_id)
         return self.get(url)
 
-    def releases(self):
-        return self.streams()
+    def get_releases(self):
+        return self.get_streams()
 
-    def themes(self):
+    def get_themes(self):
         url = '/rest/roadmap/1.0/plans/{0}/themes.json'.format(self.plan_id)
         return self.get(url)
 
-    def state(self, plan_version):
-        url = '/rest/roadmap/1.0/scheduling/{0}/state.json?planVersion={1}'.format(self.plan_id, plan_version)
+    def get_state(self):
+        url = '/rest/roadmap/1.0/scheduling/{0}/state.json'.format(self.plan_id)
         return self.get(url)
 
-    def filter(self, plan_version, limit=500):
-        url = '/rest/roadmap/1.0/plans/{0}/workitems/filter.json?planVersion={1}'.format(self.plan_id, plan_version)
+    def get_filter(self, limit=500):
+        url = '/rest/roadmap/1.0/plans/{0}/workitems/filter.json'.format(self.plan_id)
         return self.post(url, data={'limit': limit})
 
-    def filters(self, query_string):
+    def get_filters(self, query_string):
         url = '/rest/roadmap/1.0/system/filters.json?queryString={0}'.format(query_string)
         return self.get(url)
 
-    def import_issues(self, jql, limit=100, exclude_linked=True, estimation_method='estimates', epic_fetch_enabled=True,
+    def get_dependencies(self, workitem_id, plan_version):
+        url = '/rest/roadmap/1.0/workitems/{0}/dependencies.json?planVersion={1}'.format(workitem_id, plan_version)
+        return self.get(url)
+
+    def get_stage_name(self, stage_id):
+        all_stages = self.get_stages()['collection']
+        return [stage['title'] for stage in all_stages if stage['id'] == str(stage_id)][0]
+
+    def get_estimates_dict(self, estimates):
+        return {self.get_stage_name(stage['targetId']): stage['value'] for stage in estimates['stages']}
+
+    def import_workitem(self, data):
+        url = '/rest/roadmap/1.0/plans/bulk/{0}/workitems.json'.format(self.plan_id)
+        return self.post(url, data)
+
+    def get_jql_issues(self, jql, limit=500, exclude_linked=True, estimation_method='estimates', epic_fetch_enabled=True,
                       load_story_points=True):
         url = '/rest/roadmap/1.0/system/import.json'
         data = {'planId': str(self.plan_id),
@@ -78,14 +95,4 @@ def import_issues(self, jql, limit=100, exclude_linked=True, estimation_method='
                 'maxResults': limit,
                 'estimationMethod': estimation_method,
                 'loadStoryPoints': load_story_points}
-        self.post(url, data=data)
-
-    def dependencies(self, workitem_id, plan_version):
-        url = '/rest/roadmap/1.0/workitems/{0}/dependencies.json?planVersion={1}'.format(workitem_id, plan_version)
-        return self.get(url)
-
-    def stage_name(self, stage_id):
-        return [stage['title'] for stage in self.stages()['collection'] if stage['id'] == str(stage_id)][0]
-
-    def estimates_dict(self, estimates):
-        return {self.stage_name(stage['targetId']): stage['value'] for stage in estimates['stages']}
+        self.post(url, data=data)

atlassian/portfolio.py

@@ -6,9 +6,10 @@
 
 setup(
     name='atlassian-python-api',
-    description='Atlassian Python API',
+    description='Python Atlassian REST API Wrapper',
+    long_description='Python Atlassian REST API Wrapper',
     license='Apache License 2.0',
-    version='0.8.5',
+    version='0.9.0',
     download_url='https://github.com/MattAgile/atlassian-python-api',
 
     author='Matt Harasymczuk',
@@ -22,6 +23,7 @@
 
     zip_safe=False,
     install_requires=['requests==2.5.3'],
+    platforms='Platform Independent',
 
     classifiers=[
         'Development Status :: 4 - Beta',