diff --git a/testrail.py b/testrail.py index dec5d967..e123dec9 100644 --- a/testrail.py +++ b/testrail.py @@ -34,7 +34,8 @@ def __init__(self): # API: Milestones # https://mozilla.testrail.io/index.php?/api/v2/get_milestones/59 def milestones(self, testrail_project_id): - return self.client.send_get('get_milestones/{0}'.format(testrail_project_id)) + return self.client.send_get( + 'get_milestones/{0}'.format(testrail_project_id)) # API: Projects def projects(self): @@ -238,6 +239,7 @@ def testrail_milestones(self, project='all'): self.db.report_milestones_insert(projects_id, df_selected) + class DatabaseTestRail(Database): def __init__(self): @@ -266,7 +268,8 @@ def report_milestones_insert(self, projects_id, payload): for index, row in payload.iterrows(): print(row) - report = ReportTestRailMilestones(testrail_milestone_id=row['testrail_milestone_id'], + report = ReportTestRailMilestones( + testrail_milestone_id=row['testrail_milestone_id'], # noqa projects_id=projects_id, name=row['name'], started_on=row['started_on'],