Commit 4be2f277332e662ae241e49b854820431a8a8a11

Authored by Gust
Committed by Sergio Oliveira
1 parent 939a9bad

Add logger

Showing 1 changed file with 8 additions and 5 deletions   Show diff stats
colab/proxy/gitlab/data_api.py
1 import json 1 import json
2 import urllib 2 import urllib
3 import urllib2 3 import urllib2
  4 +import logging
4 5
5 from dateutil.parser import parse 6 from dateutil.parser import parse
6 7
@@ -11,6 +12,8 @@ from colab.proxy.gitlab.models import (GitlabProject, GitlabMergeRequest, @@ -11,6 +12,8 @@ from colab.proxy.gitlab.models import (GitlabProject, GitlabMergeRequest,
11 GitlabComment, GitlabIssue) 12 GitlabComment, GitlabIssue)
12 from colab.proxy.utils.proxy_data_api import ProxyDataAPI 13 from colab.proxy.utils.proxy_data_api import ProxyDataAPI
13 14
  15 +LOGGER = logging.getLogger('colab.plugin.gitlab')
  16 +
14 17
15 class GitlabDataAPI(ProxyDataAPI): 18 class GitlabDataAPI(ProxyDataAPI):
16 19
@@ -34,7 +37,7 @@ class GitlabDataAPI(ProxyDataAPI): @@ -34,7 +37,7 @@ class GitlabDataAPI(ProxyDataAPI):
34 data = urllib2.urlopen(url, timeout=10) 37 data = urllib2.urlopen(url, timeout=10)
35 json_data = json.load(data) 38 json_data = json.load(data)
36 except urllib2.URLError: 39 except urllib2.URLError:
37 - print "Connection timeout: " + url 40 + LOGGER.exception("Connection timeout: " + url)
38 json_data = [] 41 json_data = []
39 42
40 return json_data 43 return json_data
@@ -177,22 +180,22 @@ class GitlabDataAPI(ProxyDataAPI): @@ -177,22 +180,22 @@ class GitlabDataAPI(ProxyDataAPI):
177 return all_comments 180 return all_comments
178 181
179 def fetch_data(self): 182 def fetch_data(self):
180 - print "projects" 183 + LOGGER.info("Importing Projects")
181 projects = self.fetch_projects() 184 projects = self.fetch_projects()
182 for datum in projects: 185 for datum in projects:
183 datum.save() 186 datum.save()
184 187
185 - print "MR" 188 + LOGGER.info("Importing Merge Requests")
186 merge_request_list = self.fetch_merge_request(projects) 189 merge_request_list = self.fetch_merge_request(projects)
187 for datum in merge_request_list: 190 for datum in merge_request_list:
188 datum.save() 191 datum.save()
189 192
190 - print "issues" 193 + LOGGER.info("Importing Issues")
191 issue_list = self.fetch_issue(projects) 194 issue_list = self.fetch_issue(projects)
192 for datum in issue_list: 195 for datum in issue_list:
193 datum.save() 196 datum.save()
194 197
195 - print "comments" 198 + LOGGER.info("Importing Comments")
196 comments_list = self.fetch_comments() 199 comments_list = self.fetch_comments()
197 for datum in comments_list: 200 for datum in comments_list:
198 datum.save() 201 datum.save()