def push_event_hook(): push_event = request.json app.logger.debug(push_event) user_name = push_event['user_name'] repo_name = push_event['repository']['name'] repo_url = push_event['repository']['url'] repo_homepage = push_event['repository']['homepage'] refspec = push_event['ref'] app.logger.debug('Received push event by %s in branch %s on repository %s', user_name, refspec, repo_url) for commit in push_event['commits']: app.logger.debug('Processing commit %s by %s (%s) in %s', commit['id'], commit['author']['name'], commit['author']['email'], commit['url']) commit_time = dateutil.parser.parse(commit['timestamp']) issues = re.findall(app.config['REGEX'], commit['message'], re.MULTILINE) if not issues: app.logger.debug( '''Didn't find any referenced issues in commit %s''', commit['id']) else: app.logger.debug('Found %d referenced issues in commit %s', len(issues), commit['id']) yt = Connection(app.config['YOUTRACK_URL'], app.config['YOUTRACK_USERNAME'], app.config['YOUTRACK_PASSWORD']) user_login = get_user_login(yt, commit['author']['email']) if user_login is None: app.logger.warn( "Couldn't find user with email address %s. Using default user.", commit['author']['email']) default_user = yt.getUser(app.config['DEFAULT_USER']) user_login = default_user['login'] for issue_id in issues: app.logger.debug('Processing reference to issue %s', issue_id) try: yt.getIssue(issue_id) comment_string = 'Commit [%(url)s %(id)s] on branch %(refspec)s in [%(repo_homepage)s %(repo_name)s] made by %(author)s on %(date)s\n{quote}%(message)s{quote}' % { 'url': commit['url'], 'id': commit['id'], 'author': commit['author']['name'], 'date': str(commit_time), 'message': commit['message'], 'repo_homepage': repo_homepage, 'repo_name': repo_name, 'refspec': refspec } app.logger.debug(comment_string) yt.executeCommand(issueId=issue_id, command='comment', comment=comment_string.encode('utf-8'), run_as=user_login.encode('utf-8')) except YouTrackException: app.logger.warn("Couldn't find issue %s", issue_id) return Response('Push event processed. Thanks!', mimetype='text/plain')
def process_commits(secret=None): if context_secret and secret != context_secret: abort(403) yt = Connection(yt_url, yt_login, yt_password) try: cmd_pattern = re.compile( r'#((?:%s)-\d+)(?:\s+(.+))?' % '|'.join(yt.getProjects().keys()), re.IGNORECASE | re.MULTILINE) except YouTrackException: app.logger.warning('Cannot get projects from YT') cmd_pattern = re.compile(r'#([A-z]+-\d+)(?:\s+(.+))?', re.MULTILINE) payload = json.loads(request.form.get('payload')) commits_url_template = get_commits_url_template(payload) for commit in payload['commits']: message = commit['message'].encode('utf-8') issue_refs = cmd_pattern.findall(message) if not issue_refs: continue commit_node = commit['node'] commit_url = commits_url_template % commit['raw_node'] timestamp = commit['utctimestamp'] author = commit['author'].encode('utf-8') match = re.search(r'<(.+?)>', commit['raw_author']) if not match: app.logger.error("Cannot get author's email address.") abort(400) users = yt.getUsers(params={'q': match.group(1)}) if not users: app.logger.error('Cannot find user with email ' + match.group(1)) abort(400) if len(users) != 1: app.logger.error('Not unique email address ' + match.group(1)) abort(400) comment = "Commit [%s %s] made by '''%s''' on ''%s''\n{quote}%s{quote}" \ % (commit_url, commit_node, author, timestamp, message) cmd_exec_result = True for issue_id, command in issue_refs: if command is None: command = '' try: app.logger.info( "Adding commit %s to issue %s (command: %s)" % (commit_node, issue_id, command)) yt.executeCommand(issue_id, command, comment, run_as=users[0].login) except YouTrackException as e: cmd_exec_result = False app.logger.error( 'Failed to add commit %s to issue %s: %s' % (commit_node, issue_id, e.message)) if not cmd_exec_result: abort(500) return 'success'
def submit(self): try: connection = Connection(settings.YOUTRACK_URL, settings.YOUTRACK_LOGIN, settings.YOUTRACK_PASSWORD) response, content = connection.createIssue(self.project, assignee=None, summary=u'Issue from feedback form', description=self.cleaned_data['description']) print response issue_id = response['location'].split('/')[-1] connection.executeCommand(issue_id, 'Customer email ' + self.cleaned_data['email']) return True except YouTrackException: return False
def push_event_hook(room=None): push_event = request.json app.logger.debug(push_event) app.logger.debug("slack_room:%s" % room) # process mergerequest if 'object_kind' in push_event and push_event['object_kind'] == 'merge_request': obj = push_event['object_attributes'] user = push_event['user'] slack = Slacker(app.config['SLACK_TOKEN']) url = "%s/%s/%s/merge_requests/%d" % (app.config['GITLAB_HOME'], obj['target']['namespace'], obj['target']['name'], obj['iid']) print url event = "Pull Request(%s): %s - %s\n%s" % (obj['state'], url, obj['title'], obj['description']) slack.chat.post_message('#%s' % room, event, username=user['username'], icon_url=user['avatar_url']) return Response('Push event processed. Thanks!', mimetype='text/plain') user_name = push_event['user_name'] repo_name = push_event['repository']['name'] repo_url = push_event['repository']['url'] repo_homepage = push_event['repository']['homepage'] refspec = push_event['ref'] app.logger.debug('Received push event by %s in branch %s on repository %s', user_name, refspec, repo_url) for commit in push_event['commits']: app.logger.debug('Processing commit %s by %s (%s) in %s', commit['id'], commit['author']['name'], commit['author']['email'], commit['url']) commit_time = dateutil.parser.parse(commit['timestamp']) issues = re.findall(app.config['REGEX'], commit['message'], re.MULTILINE) if not issues: app.logger.debug('''Didn't find any referenced issues in commit %s''', commit['id']) else: app.logger.debug('Found %d referenced issues in commit %s', len(issues), commit['id']) yt = Connection(app.config['YOUTRACK_URL'], app.config['YOUTRACK_USERNAME'], app.config['YOUTRACK_PASSWORD']) user_login = get_user_login(yt, commit['author']['email']) if user_login is None: app.logger.warn("Couldn't find user with email address %s. Using default user.", commit['author']['email']) default_user = yt.getUser(app.config['DEFAULT_USER']) user_login = default_user['login'] for issue_id in issues: app.logger.debug('Processing reference to issue %s', issue_id) try: yt.getIssue(issue_id) comment_string = 'Commit [%(url)s %(id)s] on branch %(refspec)s in [%(repo_homepage)s %(repo_name)s] made by %(author)s on %(date)s\n{quote}%(message)s{quote}' % {'url': commit['url'], 'id': commit['id'], 'author': commit['author']['name'], 'date': str(commit_time), 'message': commit['message'], 'repo_homepage': repo_homepage, 'repo_name': repo_name, 'refspec': refspec} app.logger.debug(comment_string) yt.executeCommand(issueId=issue_id, command='comment', comment=comment_string.encode('utf-8'), run_as=user_login.encode('utf-8')) except YouTrackException: app.logger.warn("Couldn't find issue %s", issue_id) return Response('Push event processed. Thanks!', mimetype='text/plain')
def add_tags(tag_file,target_url, target_login, target_password): """ Add tags from the tag file (run after known creation) :param tag_file: :return: """ target = Connection(target_url, target_login, target_password) with open(tag_file) as tag_o_file: reader = csv.reader(tag_o_file) for row in reader: issue = row[0] for t in row[1:]: print "executing tag for %s" %(issue) target.executeCommand(issue, "tag %s" % (t))
def process_commits(secret=None): if context_secret and secret != context_secret: abort(403) yt = Connection(yt_url, yt_login, yt_password) try: cmd_pattern = re.compile( r'#((?:%s)-\d+)(?:\s+(.+))?' % '|'.join(yt.getProjects().keys()), re.IGNORECASE | re.MULTILINE) except YouTrackException: app.logger.warning('Cannot get projects from YT') cmd_pattern = re.compile(r'#([A-z]+-\d+)(?:\s+(.+))?', re.MULTILINE) payload = json.loads(request.form.get('payload')) commits_url_template = get_commits_url_template(payload) for commit in payload['commits']: message = commit['message'].encode('utf-8') issue_refs = cmd_pattern.findall(message) if not issue_refs: continue commit_node = commit['node'] commit_url = commits_url_template % commit['raw_node'] timestamp = commit['utctimestamp'] author = commit['author'].encode('utf-8') match = re.search(r'<(.+?)>', commit['raw_author']) if not match: app.logger.error("Cannot get author's email address.") abort(400) users = yt.getUsers(params={'q': match.group(1)}) if not users: app.logger.error('Cannot find user with email ' + match.group(1)) abort(400) if len(users) != 1: app.logger.error('Not unique email address ' + match.group(1)) abort(400) comment = "Commit [%s %s] made by '''%s''' on ''%s''\n{quote}%s{quote}" \ % (commit_url, commit_node, author, timestamp, message) cmd_exec_result = True for issue_id, command in issue_refs: if command is None: command = '' try: app.logger.info("Adding commit %s to issue %s (command: %s)" % (commit_node, issue_id, command)) yt.executeCommand(issue_id, command, comment, run_as=users[0].login) except YouTrackException as e: cmd_exec_result = False app.logger.error('Failed to add commit %s to issue %s: %s' % (commit_node, issue_id, e.message)) if not cmd_exec_result: abort(500) return 'success'
def submit(self): try: connection = Connection(settings.YOUTRACK_URL, settings.YOUTRACK_LOGIN, settings.YOUTRACK_PASSWORD) response, content = connection.createIssue(self.project, assignee=None, summary=self.get_summary().encode('utf-8'), description=self.cleaned_data['description'].encode('utf-8')) issue_id = response['location'].split('/')[-1] commands = '' if self.subsystem is not None: commands += ' Subsystem %s' % self.subsystem commands += ' Customer email ' + self.cleaned_data['email'] connection.executeCommand(issue_id, commands) return True except YouTrackException: return False
def submit(self): try: connection = Connection(settings.YOUTRACK_URL, settings.YOUTRACK_LOGIN, settings.YOUTRACK_PASSWORD) response, content = connection.createIssue( self.project, assignee=None, summary=u'Issue from feedback form', description=self.cleaned_data['description']) print response issue_id = response['location'].split('/')[-1] connection.executeCommand( issue_id, 'Customer email ' + self.cleaned_data['email']) return True except YouTrackException: return False
class YtClient(object): def __init__(self, server, username, password): self.logger = LoggerFactory.getLogger("com.xebialabs.yourtrack.Server") if server is None: sys.exit("No server provided.") if username is not None: youtrackUsername = username else: youtrackUsername = server['username'] if password is not None: youtrackPassword = password else: youtrackPassword = server['password'] if server['token'] is not None: # authentication request with permanent token self.youtrack = YouTrack(server['url'], token=server['token']) else: # authentication with username and password self.youtrack = YouTrack(server['url'], login=youtrackUsername, password=youtrackPassword) def getIssue(self, issueId): return self.youtrack.getIssue(issueId) def getAllIssues(self, query, withFields): self.logger.debug("getAllIssues") self.logger.debug("Query = %s" % query) self.logger.debug("Fields = %s" % withFields) return self.youtrack.getAllIssues(filter=query, withFields=withFields) def updateIssuesByQuery(self, query, fieldsToUpdate, comment): command = "" for key in fieldsToUpdate: command += key + " " + fieldsToUpdate[key] foundIssues = self.youtrack.getAllIssues(filter=query, withFields=[]) for issue in foundIssues: self.youtrack.executeCommand(issue['id'], command, comment) def getIssuesByQuery(self, query, fieldList=[]): self.logger.debug("getIssuesByQuery") return self.youtrack.getAllIssues(filter=query, withFields=[])
def push_event_hook(): push_event = request.json app.logger.debug(push_event) user_name = push_event['user_name'] repo_name = push_event['repository']['name'] repo_url = push_event['repository']['url'] repo_homepage = push_event['repository']['homepage'] refspec = push_event['ref'] app.logger.debug('Received push event by %s in branch %s on repository %s', user_name, refspec, repo_url) for commit in push_event['commits']: app.logger.debug('Processing commit %s by %s (%s) in %s', commit['id'], commit['author']['name'], commit['author']['email'], commit['url']) commit_time = dateutil.parser.parse(commit['timestamp']) issues = re.findall(app.config['REGEX'], commit['message'], re.MULTILINE) if not issues: app.logger.debug('''Didn't find any referenced issues in commit %s''', commit['id']) else: app.logger.debug('Found %d referenced issues in commit %s', len(issues), commit['id']) yt = Connection(app.config['YOUTRACK_URL'], app.config['YOUTRACK_USERNAME'], app.config['YOUTRACK_PASSWORD']) default_user = yt.getUser(app.config['DEFAULT_USER']) user_login = default_user['login'] users = yt.getUsers({ 'q': commit['author']['email'] }) if not users: app.logger.warn('''Couldn't find user with email address %s. Using default user.''', commit['author']['email']) elif len(users) > 1: app.logger.warn('''Found more than one user with email address %s. Using default user.''', commit['author']['email']) else: user_login = users[0]['login'] for issue_id in issues: app.logger.debug('Processing reference to issue %s', issue_id) try: issue = yt.getIssue(issue_id) comment_string = 'Commit [%(url)s %(id)s] on branch %(refspec)s in [%(repo_homepage)s %(repo_name)s] made by %(author)s on %(date)s\n{quote}%(message)s{quote}' % {'url': commit['url'], 'id': commit['id'], 'author': commit['author']['name'], 'date': str(commit_time), 'message': commit['message'], 'repo_homepage': repo_homepage, 'repo_name': repo_name, 'refspec': refspec} app.logger.debug(comment_string) yt.executeCommand(issueId=issue_id, command='comment', comment=comment_string.encode('utf-8'), run_as=user_login.encode('utf-8')) except YouTrackException: app.logger.warn('''Couldn't find issue %s''', ref) return Response('Push event processed. Thanks!', mimetype='text/plain')
def issue_actualizar_estado(bot, update, user_data): bot.sendChatAction(chat_id=update.callback_query.from_user.id, action=ChatAction.TYPING) if update.callback_query.data == 'issue_estado_cerrar': estado = 'Fixed' else: estado = 'In Progress' logger.info('Actualizar estado issue {} {}'.format(user_data['issue'], estado)) try: connection = Connection(user_data['host']['host'], user_data['host']['username'], user_data['host']['pass']) command = 'State ' + estado connection.executeCommand(user_data['issue'], command) update.callback_query.edit_message_text("Gracias amego!") except Exception as e: logger.error(e) update.callback_query.edit_message_text("No eh!") return ConversationHandler.END
def receive_hook(): payload = request.json user_name = payload['user_name'] repo_url = payload['repository']['url'] app.logger.debug('Received payload for a push by %s on repository %s', user_name, repo_url) for commit in payload['commits']: app.logger.debug('Processing commit %s by %s (%s) in %s', commit['id'], commit['author']['name'], commit['author']['email'], commit['url']) commit_time = dateutil.parser.parse(commit['timestamp']) refs = re.findall(app.config['REGEX'], commit['message'], re.MULTILINE) if not refs: app.logger.info('''Didn't find any referenced issues in commit %s''', commit['id']) else: app.logger.info('Found %d referenced issues in commit %s', len(refs), commit['id']) yt = Connection(app.config['YOUTRACK_URL'], app.config['YOUTRACK_USERNAME'], app.config['YOUTRACK_PASSWORD']) user = app.config['DEFAULT_USER'] users = yt.getUsers({ 'q': commit['author']['email'] }) if not users: app.logger.warn('''Couldn't find user with email address %s. Using default user.''', commit['author']['email']) elif len(users) > 1: app.logger.warn('''Found more than one user with email address %s. Using default user.''', commit['author']['email']) else: user = users[0]['login'] for ref in refs: app.logger.info('Processing reference to issue %s', ref) try: issue = yt.getIssue(ref) comment_string = 'Commit [%(url)s %(id)s] made by %(author)s on %(date)s\n{quote}%(message)s{quote}' % {'url': commit['url'], 'id': commit['id'], 'author': commit['author']['name'], 'date': str(commit_time), 'message': commit['message']} app.logger.debug(comment_string) yt.executeCommand(issueId=ref, command='comment', comment=comment_string, run_as=user) except YouTrackException: app.logger.warn('''Couldn't find issue %s''', ref) return Response('Payload processed. Thanks!', mimetype='text/plain')
if params.get('add_new_comments'): target_comments = dict() max_id = 0 for c in target_issue.getComments(): target_comments[c.created] = c if max_id < c.created: max_id = c.created for c in issue.getComments(): if c.created > max_id or c.created not in target_comments: group = None if hasattr(c, 'permittedGroup'): group = c.permittedGroup try: target.executeCommand( issue.id, 'comment', c.text, group, c.author) except youtrack.YouTrackException, e: print 'Cannot add comment to issue ' print e if params.get('sync_custom_fields'): skip_fields = [] if tt_settings and tt_settings.Enabled and tt_settings.TimeSpentField: skip_fields.append(tt_settings.TimeSpentField) skip_fields = [name.lower() for name in skip_fields] for pcf in [ pcf for pcf in project_custom_fields if pcf.name.lower() not in skip_fields ]: target_cf_value = None
for issue in issues: try: target_issue = target.getIssue(issue.id) except youtrack.YouTrackException, e: print "Cannot get target issue" print e continue if params.get('sync_tags') and issue.tags: try: for tag in issue.tags: tag = re.sub(r'[,&<>]', '_', tag) try: target.executeCommand( issue.id, 'tag ' + tag, disable_notifications=True) except youtrack.YouTrackException: tag = re.sub(r'[\s-]', '_', tag) target.executeCommand( issue.id, 'tag ' + tag, disable_notifications=True) except youtrack.YouTrackException, e: print "Cannot sync tags for issue " + issue.id print e if params.get('add_new_comments'): target_comments = dict() max_id = 0 for c in target_issue.getComments():
def trac2youtrack(target_url, target_login, target_password, project_ID, project_name, env_path): # creating connection to trac to import issues to client = Client(env_path) # creating connection to util to import issues in target = Connection(target_url, target_login, target_password) # create project print("Creating project[%s]" % project_name) try: target.getProject(project_ID) except youtrack.YouTrackException: target.createProjectDetailed(project_ID, project_name, client.get_project_description(), target_login) # importing users trac_users = client.get_users() print("Importing users") yt_users = list([]) # converting trac users to yt users registered_users = set([]) for user in trac_users : print("Processing user [ %s ]" % user.name) registered_users.add(user.name) yt_users.append(to_youtrack_user(user)) # adding users to yt project target.importUsers(yt_users) print("Importing users finished") print("Creating project custom fields") create_yt_custom_field(target, project_ID, "Priority", client.get_issue_priorities()) create_yt_custom_field(target, project_ID, "Type", client.get_issue_types()) trac_resolution_to_yt_state = lambda track_field, yt_bundle : to_youtrack_state(track_field, yt_bundle) create_yt_bundle_custom_field(target, project_ID, "Resolution", client.get_issue_resolutions(), trac_resolution_to_yt_state) trac_version_to_yt_version = lambda trac_field, yt_bundle : to_youtrack_version(trac_field, yt_bundle) trac_versions = client.get_versions() create_yt_bundle_custom_field(target, project_ID, "Affected versions", trac_versions, trac_version_to_yt_version) trac_milestones = client.get_milestones() create_yt_bundle_custom_field(target, project_ID, "Fix versions", trac_milestones, trac_version_to_yt_version) trac_components = client.get_components() for cmp in trac_components : if cmp.owner not in registered_users : cmp.owner, registered_users = process_non_authorised_user(target, registered_users, cmp.owner) trac_component_to_yt_subsystem = lambda trac_field, yt_bundle : to_youtrack_subsystem(trac_field, yt_bundle) create_yt_bundle_custom_field(target, project_ID, "Component", trac_components, trac_component_to_yt_subsystem) create_yt_custom_field(target, project_ID, "Severity", client.get_severities()) trac_custom_fields = client.get_custom_fields_declared() check_box_fields = dict([]) for elem in trac_custom_fields: print("Processing custom field [ %s ]" % elem.name) if elem.type == "checkbox": if len(elem.label) > 0: opt = elem.label else: opt = elem.name options = list([opt]) check_box_fields[elem.name] = opt else: options = elem.options values = None if len(options): values = options field_name = elem.name if field_name in youtrackutils.tracLib.FIELD_NAMES.keys() : field_name = youtrackutils.tracLib.FIELD_NAMES[field_name] field_type = youtrackutils.tracLib.CUSTOM_FIELD_TYPES[elem.type] if field_name in youtrackutils.tracLib.FIELD_TYPES.keys(): field_type = youtrackutils.tracLib.FIELD_TYPES[field_name] process_custom_field(target, project_ID, field_type, field_name, trac_values_to_youtrack_values(field_name, values)) print("Creating project custom fields finished") print("Importing issues") trac_issues = client.get_issues() yt_issues = list([]) counter = 0 max = 100 for issue in trac_issues: print("Processing issue [ %s ]" % (str(issue.id))) counter += 1 if not (issue.reporter in registered_users): yt_user, registered_users = process_non_authorised_user(target, registered_users, issue.reporter) if yt_user is None : issue.reporter = "guest" else: issue.reporter = yt_user if not (issue.owner in registered_users): yt_user, registered_users = process_non_authorised_user(target, registered_users, issue.owner) if yt_user is None : issue.owner = "" else: issue.owner = yt_user legal_cc = set([]) for cc in issue.cc: if cc in registered_users: legal_cc.add(cc) issue.cc = legal_cc yt_issues.append(to_youtrack_issue(project_ID, issue, check_box_fields)) if counter == max: counter = 0 print(target.importIssues(project_ID, project_name + ' Assignees', yt_issues)) yt_issues = list([]) print(target.importIssues(project_ID, project_name + ' Assignees', yt_issues)) print('Importing issues finished') # importing tags print("Importing keywords") for issue in trac_issues: print("Importing tags from issue [ %s ]" % (str(issue.id))) tags = issue.keywords for t in tags: target.executeCommand(str(project_ID) + "-" + str(issue.id), "tag " + t.encode('utf-8')) print("Importing keywords finished") print("Importing attachments") for issue in trac_issues: print("Processing issue [ %s ]" % (str(issue.id))) issue_attach = issue.attachment for attach in issue_attach: print("Processing attachment [ %s ]" % attach.filename.encode('utf-8')) if not (attach.author_name in registered_users): yt_user, registered_users = process_non_authorised_user(target, registered_users, attach.author_name) if yt_user is None: attach.author_name = "guest" else: attach.author_name = yt_user content = open(urllib.quote(attach.filename.encode('utf-8'))) target.createAttachment(str(project_ID) + "-" + str(issue.id), attach.name, content, attach.author_name, created=attach.time) print("Importing attachments finished") print("Importing workitems") tt_enabled = False for issue in trac_issues: if issue.workitems: if not tt_enabled: tt_settings = target.getProjectTimeTrackingSettings(str(project_ID)) if not tt_settings.Enabled: print("Enabling TimeTracking for the project") target.setProjectTimeTrackingSettings(str(project_ID), enabled=True) tt_enabled = True print("Processing issue [ %s ]" % (str(issue.id))) workitems = [to_youtrack_workitem(w) for w in issue.workitems] target.importWorkItems(str(project_ID) + "-" + str(issue.id), workitems) print("Importing workitems finished")
def mantis2youtrack( target_url, target_login, target_pass, mantis_db_name, mantis_db_host, mantis_db_port, mantis_db_login, mantis_db_pass, mantis_project_names, ): print "target_url : " + target_url print "target_login : "******"target_pass : "******"mantis_db_name : " + mantis_db_name print "mantis_db_host : " + mantis_db_host print "mantis_db_port : " + mantis_db_port print "mantis_db_login : "******"mantis_db_pass : "******"mantis_project_names : " + repr(mantis_project_names) # connacting to yt target = Connection(target_url, target_login, target_pass) # connacting to mantis client = MantisClient( mantis_db_host, int(mantis_db_port), mantis_db_login, mantis_db_pass, mantis_db_name, mantis.CHARSET ) if not len(mantis_project_names): print "You should declarer at least one project to import" sys.exit() value_sets = dict([]) print "Importing users" yt_users = [] value_sets["user"] = set([]) for user in client.get_mantis_users(): print "Processing user [ %s ]" % user.user_name value_sets["user"].add(user.user_name) yt_users.append(to_yt_user(user)) target.importUsers(yt_users) print "Importing users finished" print "Creating custom fields definitions" value_sets.update(create_custom_fields(target, u"priority", mantis.PRIORITY_VALUES.values())) value_sets.update(create_custom_fields(target, u"severity", mantis.SEVERITY_VALUES.values())) value_sets.update(create_custom_fields(target, u"category", ["No subsystem"], "1")) value_sets.update(create_custom_fields(target, u"version", [], "1")) value_sets.update(create_custom_fields(target, u"fixed_in_version", [], "1")) value_sets.update(create_custom_fields(target, u"build", [], "1")) value_sets.update(create_custom_fields(target, u"platform")) value_sets.update(create_custom_fields(target, u"os")) value_sets.update(create_custom_fields(target, u"os_build")) value_sets.update(create_custom_fields(target, u"due_date")) value_sets.update(create_custom_fields(target, u"Reproducibility", mantis.REPRODUCIBILITY_VALUES.values())) # create custom field for target version field = None try: field = target.getCustomField("Fix versions") except YouTrackException: pass if field is not None: if hasattr(field, "defaultBundle"): bundle = field.defaultBundle for name in get_yt_cf_name_from_mantis_cf_name("target_version"): try: target.createCustomFieldDetailed( name, mantis.FIELD_TYPES[name], False, True, True, {"defaultBundle": bundle, "attachBundlePolicy": "1"}, ) except YouTrackException: pass value_sets.update( create_auto_attached_bundle_custom_fields( target, u"status", mantis.STATUS_VALUES.values(), lambda status, bundle, value_mapping: to_yt_state(status, bundle, value_mapping), ) ) value_sets.update( create_auto_attached_bundle_custom_fields( target, u"resolution", mantis.RESOLUTION_VALUES.values(), lambda resolution, bundle, value_mapping: to_yt_state(resolution, bundle, value_mapping), ) ) if mantis.CREATE_CF_FOR_SUBPROJECT: value_sets.update(create_custom_fields(target, u"subproject", [], "1")) handler_field_name = u"handler" value_sets.update(create_custom_fields(target, handler_field_name, [], "1")) for name in get_yt_cf_name_from_mantis_cf_name(handler_field_name): value_sets[name] = value_sets["user"] # adding some custom fields that are predefined in mantis project_ids = [] for name in mantis_project_names: project_ids.append(client.get_project_id_by_name(name)) custom_fields = client.get_mantis_custom_fields(project_ids) for cf_def in custom_fields: print "Processing custom field [ %s ]" % cf_def.name.encode("utf-8") value_sets.update(process_mantis_custom_field(target, cf_def)) print "Creating custom fields definitions finished" for name in mantis_project_names: project_id = int(client.get_project_id_by_name(name)) print "Creating project [ %s ] with name [ %s ]" % (project_id, name) try: target.getProject(str(project_id)) except YouTrackException: target.createProjectDetailed( str(project_id), name, client.get_project_description(project_id), target_login ) print "Importing components to project [ %s ]" % project_id value_sets.update( add_values_to_fields( target, project_id, u"category", client.get_mantis_categories(project_id), lambda component, yt_bundle, value_mapping: to_yt_subsystem(component, yt_bundle, value_mapping), ) ) print "Importing components to project [ %s ] finished" % project_id print "Importing versions to project [ %s ]" % project_id mantis_versions = client.get_mantis_versions(project_id) value_sets.update( add_values_to_fields( target, project_id, u"version", mantis_versions, lambda version, yt_bundle, value_mapping: to_yt_version(version, yt_bundle, value_mapping), ) ) value_sets.update( add_values_to_fields( target, project_id, u"fixed_in_version", mantis_versions, lambda version, yt_bundle, value_mapping: to_yt_version(version, yt_bundle, value_mapping), ) ) for name in get_yt_cf_name_from_mantis_cf_name("target_version"): value_sets[name] = value_sets["Fix versions"] print "Importing versions to project [ %s ] finished" % project_id print "Attaching custom fields to project [ %s ]" % project_id cf_ids = client.get_custom_fields_attached_to_project(project_id) ignore = [] for cf in custom_fields: if cf.field_id in cf_ids: attach_field_to_project(target, project_id, cf.name) else: ignore.append(cf.name) if mantis.CREATE_CF_FOR_SUBPROJECT: value_sets.update( add_values_to_fields( target, project_id, u"subproject", client.get_mantis_subprojects(project_id), lambda sp_name, yt_bundle, value_mapping: yt_bundle.createElement(sp_name) if sp_name not in value_mapping else yt_bundle.createElement(value_sets[sp_name]), ) ) print "Attaching custom fields to project [ %s ] finished" % project_id print "Importing issues to project [ %s ]" % project_id mantis_issues = client.get_mantis_issues(project_id) yt_issues = [] max_count = 100 for issue in mantis_issues: # print "Processing issue [ %s ]" % str(issue.id) yt_issues.append(to_yt_issue(issue, value_sets, ignore)) if len(yt_issues) >= max_count: print target.importIssues(str(project_id), name + "Assignees", yt_issues) yt_issues = [] target.importIssues(str(project_id), str(project_id) + "Assignees", yt_issues) print "Importing issues to project [ %s ] finished" % project_id print "Importing issue attachments to project [ %s ]" % project_id mantis_attachments = client.get_mantis_attachments(project_id) for attachment in mantis_attachments: print "Processing issue attachment [ %s ]" % str(attachment.id) content = StringIO(attachment.content) authorLogin = client.get_user_name_by_id(attachment.user_id) target.createAttachment( "%s-%s" % (project_id, attachment.bug_id), attachment.filename, content, authorLogin, attachment.file_type, None, str(attachment.date_added * 1000), ) print "Importing issue attachments to project [ %s ] finished" % project_id print "Importing tags to issues from project [ %s ]" % project_id for issue in mantis_issues: print "Processing tags for issue [ %s ]" % str(issue.id) for tag in issue.tags: print "Processing tag [ %s ]" % tag.encode("utf8") target.executeCommand(str(project_id) + "-" + str(issue.id), "tag " + tag.encode("utf8")) print "Importing tags to issues from project [ %s ] finished" % project_id print "Importing issue links" mantis_issue_links = client.get_issue_links() yt_issue_links = [] for link in mantis_issue_links: print "Processing issue link for source issue [ %s ]" % str(link.source) yt_issue_links.append(to_yt_link(link)) print target.importLinks(yt_issue_links) print "Importing issue links finished"
class RedmineImporter(object): def __init__(self, params): self._params = params self._projects = None self._max_issue_ids = {} self._issue_ids = {} self._relations = {} self._users = {} self._groups = {} self._subsystems = {} self._versions = {} self._tags = {} self._all_users_group = None # Connecting to Redmine rm_api_key = params.get('rm_api_key') if not rm_api_key and 'rm_api_key_file' in params: try: with open(params['rm_api_key_file'], 'r') as f: rm_api_key = f.read().strip() except (OSError, IOError) as e: print(("Cannot load redmine api key from file: " + str(e))) sys.exit(1) if rm_api_key: self._source = youtrackutils.redmine.RedmineClient( rm_api_key, params['rm_url']) elif 'rm_login' in params: self._source = youtrackutils.redmine.RedmineClient( None, params['rm_url'], params.get('rm_login'), params.get('rm_password')) else: print("You have to provide Redmine API key or login/password") sys.exit(1) # Connecting to YouTrack token = params.get('token') if not token and 'token_file' in params: try: with open(params['token_file'], 'r') as f: token = f.read().strip() except (OSError, IOError) as e: print(("Cannot load token from file: " + str(e))) sys.exit(1) if token: self._target = Connection(params['yt_url'], token=token) elif 'yt_login' in params: self._target = Connection(params['yt_url'], params.get('yt_login'), params.get('yt_password')) else: print("You have to provide YouTrack token or login/password") sys.exit(1) if not params.get('project_lead_login'): project_lead = params.get('yt_login') if not project_lead: for login in ('root', 'admin', 'administrator', 'guest'): try: project_lead = self._target.getUser(login).login break except youtrack.YouTrackException: continue self._params['project_lead_login'] = project_lead def do_import(self, project_ids): try: projects2import = self._get_projects(project_ids) except youtrackutils.redmine.RedmineException as e: print(('FATAL:', e)) sys.exit(1) print('===> Import Roles') self._import_roles() for project in list(projects2import.values()): self._import_project(project) print('===> Apply Relations') self._apply_relations() def _get_projects(self, project_ids=None, by_internal_id=False): if by_internal_id: by = 'by_iid' else: by = 'by_pid' if self._projects is None: self._projects = {'by_iid': {}, 'by_pid': {}} if project_ids: new_projects = [ pid for pid in project_ids if pid not in self._projects[by] ] else: new_projects = None if new_projects is None or new_projects: for project in self._source.get_projects(new_projects): project.identifier = re.sub('\W', '', project.identifier) self._projects['by_iid'][project.id] = project self._projects['by_pid'][project.identifier] = project if project_ids: for pid in project_ids: if by == 'by_pid': pid = re.sub('\W', '', pid) if pid not in self._projects[by]: raise youtrackutils.redmine.RedmineException( "Project '%s' doesn't exist in Redmine" % pid) return self._projects[by] def _get_project(self, project_id, by_internal_id=False): return self._get_projects([project_id], by_internal_id)[project_id] def _get_project_name(self, project): name = project.name while True: if not hasattr(project, 'parent'): break name = project.parent.name + ' :: ' + name project = self._get_project(project.parent.id, True) return name def _import_project(self, project): project_id = project.identifier project_name = self._get_project_name(project) project_desc = '' project_lead = self._params['project_lead_login'] if hasattr(project, 'description') and project.description is not None: project_desc = project.description print( ("===> Importing Project '%s' (%s)" % (project_name, project_id))) try: print('Creating project...') self._target.getProject(project_id) print('Project already exists') except youtrack.YouTrackException: self._target.createProjectDetailed(project_id, project_name, project_desc, project_lead) print('Project successfully created') print('Import project members...') self._import_members(project) print('Import issues...') self._import_issues(project) def _to_yt_user(self, redmine_user): if isinstance(redmine_user, str): user_id = redmine_user else: user_id = redmine_user.id if user_id not in self._users: try: redmine_user = self._source.get_user(user_id) except ResourceNotFound: redmine_user = None user = youtrack.User() if redmine_user != None: try: user.email = redmine_user.mail except AttributeError: pass if not user.email and redmine_user.id == 2: user.email = '*****@*****.**' try: # In some cases redmine user login can be empty or missing. # So, both cases should be handled. user.login = redmine_user.login except AttributeError: pass if not user.login and redmine_user.id == 2: user.login = '******' if not hasattr(user, 'login') or not user.login: if hasattr(user, 'email') and user.email: user.login = user.email else: user.login = '******' print(('Cannot get login for user id=%s, set it to "%s"' % (user_id, user.login))) # user.login = redmine_user.login or 'guest' # user.email = redmine_user.mail or '*****@*****.**' if redmine_user != None and user.login != 'guest': if redmine_user.firstname is None and redmine_user.lastname is None: user.fullName = user.login elif redmine_user.firstname is None: user.fullName = redmine_user.lastname elif redmine_user.lastname is None: user.fullName = redmine_user.firstname else: user.fullName = redmine_user.firstname + ' ' + redmine_user.lastname else: user.created = True if redmine_user != None and hasattr(redmine_user, 'groups'): user.groups = [ self._to_yt_group(g) for g in redmine_user.groups ] self._users[user_id] = user return self._users[user_id] def _to_yt_group(self, redmine_group, users=None): if not isinstance(redmine_group, str): redmine_group = redmine_group.name if redmine_group not in self._groups: group = youtrack.Group() group.name = redmine_group if users is None: users = [] group.users = users self._groups[redmine_group] = group return self._groups[redmine_group] def _to_yt_role(self, name, projects=None): role = youtrack.UserRole() role.name = name if projects: if isinstance(projects, list): role.projects.extend(projects) else: role.projects.append(projects) return role def _to_yt_version(self, version): if isinstance(version, str): vid = version else: vid = version.id if vid not in self._versions: redmine_version = self._source.get_version(vid) version = youtrack.Version() version.name = redmine_version.name version.description = redmine_version.description if getattr(redmine_version, 'due_date', None): version.releaseDate = str(to_unixtime( redmine_version.due_date)) version.released = str(redmine_version.status == 'closed').lower() version.archived = 'false' self._versions[vid] = version return self._versions[vid] def _to_yt_subsystem(self, category): if isinstance(category, str): cid = category else: cid = category.id if cid not in self._subsystems: redmine_cat = self._source.get_category(cid) subsystem = youtrack.Subsystem() subsystem.name = redmine_cat.name if hasattr(redmine_cat, 'assigned_to'): subsystem.login = self._create_user( redmine_cat.assigned_to).login self._subsystems[cid] = subsystem return self._subsystems[cid] def _get_assignee_group_name(self, project_id): return '%s Assignees' % project_id.upper() def _get_yt_issue_id(self, issue, as_number_in_project=False): project_id = self._projects['by_iid'][issue.project.id].identifier new_id = self._max_issue_ids.get(project_id, 0) + 1 rid = int(issue.id) if rid not in self._issue_ids: self._max_issue_ids[project_id] = new_id self._issue_ids[rid] = {'id': new_id, 'project_id': project_id} if as_number_in_project: return self._issue_ids[rid]['id'] return self._to_yt_issue_id(rid) def _to_yt_issue_id(self, iid): issue = self._issue_ids[iid] return '%s-%d' % (issue['project_id'], issue['id']) def _get_yt_issue_number(self, issue): return self._get_yt_issue_id(issue, True) def _import_members(self, project): project_id = project.identifier members = self._source.get_project_members(project.id) users_by_role = {} groups_by_role = {} if members: for member in members: # Sometimes roles can be duplicated if hasattr(member, 'roles'): roles = set([r.name for r in member.roles]) if hasattr(member, 'group'): group = self._to_yt_group(member.group.name) for role in roles: if role not in groups_by_role: groups_by_role[role] = [] groups_by_role[role].append(group) else: user = self._to_yt_user(member.user) for role in roles: if role not in users_by_role: users_by_role[role] = [] users_by_role[role].append(user) for role_name, users in list(users_by_role.items()): group = self._to_yt_group('%s %s' % (project_id.upper(), role_name)) self._create_group(group) self._target.addUserRoleToGroup( group, self._to_yt_role(role_name, project_id)) self._target.importUsers(users) for user in users: self._target.setUserGroup(user.login, group.name) for role_name, groups in list(groups_by_role.items()): for group in groups: self._create_group(group) self._target.addUserRoleToGroup( group, self._to_yt_role(role_name, project_id)) def _import_roles(self): existed_roles = [role.name for role in self._target.getRoles()] new_roles = {} for role in self._source.get_roles(): if role.name in existed_roles: continue permissions = None if hasattr(role, 'permissions'): permissions = [] for perm in role.permissions: if isinstance(perm, str): rm_perm = perm else: rm_perm = perm.name yt_perm = \ youtrackutils.redmine.Mapping.PERMISSIONS.get(rm_perm) if not yt_perm: continue if isinstance(yt_perm, list): permissions.extend(yt_perm) else: permissions.append(yt_perm) new_roles[role.name] = permissions for role_name, role_permissions in list(new_roles.items()): role = self._to_yt_role(role_name) self._target.createRole(role) if role_permissions: for perm_name in role_permissions: perm = youtrack.Permission() perm.name = perm_name self._target.addPermissionToRole(role, perm) def _import_issues(self, project): project_id = project.identifier limit = self._params.get('batch_size', DEFAULT_BATCH_SIZE) offset = 0 assignee_group = self._get_assignee_group_name(project_id) while True: print(("---> Issues from {offset} to {limit}".format( offset=offset + 1, limit=offset + limit))) issues = self._source.get_project_issues( project.id, limit, offset, self._params.get('skip_on_error', False)) if not issues: break issues = [ issue for issue in issues if issue.project.id == project.id ] self._target.importIssues( project_id, assignee_group, [self._make_issue(issue, project_id) for issue in issues]) for issue in issues: self._add_tags(issue) self._collect_relations(issue) self._add_attachments(issue) if self._params.get('import_time_entries', False): self._enable_timetracking(project) self._add_work_items(issue) offset += limit def _make_issue(self, redmine_issue, project_id): issue = youtrack.Issue() issue['comments'] = [] print(("--- ISSUE #{0}".format(redmine_issue.id))) if self._params.get('use_markdown'): issue['markdown'] = "true" try: if not hasattr(redmine_issue, 'author'): redmine_issue.author = 2 # set to Anonymous if self._params.get('create_redmine_linkage', False): self._add_field_to_issue(project_id, issue, "redmine_id", int(redmine_issue.id)) checklist_value = None for name, value in list(redmine_issue.attributes.items()): if name in ('project', 'attachments', 'tags'): continue if name == 'assigned_to' and value.name in self._groups: continue if name == 'id': value = str(self._get_yt_issue_number(redmine_issue)) if name == 'checklist': checklist_value = str(value) continue if name == 'custom_fields': for field in value: self._add_field_to_issue(project_id, issue, field.name, getattr(field, 'value', None)) elif name == 'journals': self._add_journals(issue, value) else: if name == 'category': value = self._to_yt_subsystem(value) self._add_field_to_issue(project_id, issue, name, value) if checklist_value: issue['description'] = issue[ 'description'] + "\n\n#### Checklist\n" + checklist_value except Exception as e: print('Failed to process issue:') print(redmine_issue) traceback.print_exc() raise e return issue def _convert_value(self, field_name, value): conv_map = youtrackutils.redmine.Mapping.CONVERSION.get(field_name) if conv_map: if hasattr(value, 'value'): if value.value in conv_map: value.value = conv_map[value.value] elif hasattr(value, 'name'): if value.name in conv_map: value.name = conv_map[value.name] return value def _get_yt_field_name(self, field_name): return youtrackutils.redmine.Mapping.FIELD_NAMES.get( field_name, field_name) def _get_yt_field_type(self, field_name): return youtrackutils.redmine.Mapping.FIELD_TYPES.get( field_name, youtrack.EXISTING_FIELD_TYPES.get(field_name)) def _add_field_to_issue(self, project_id, issue, name, value): if value is None: return field_name = self._get_yt_field_name(name) field_type = self._get_yt_field_type(field_name) if field_type is None: return value = self._convert_value(field_name, value) if isinstance(value, list): if not value: return issue[field_name] = [] for v in value: v = self._create_field_value(project_id, field_name, field_type, v) issue[field_name].append( self._get_value_presentation(field_type, v)) else: value = self._create_field_value(project_id, field_name, field_type, value) issue[field_name] = self._get_value_presentation(field_type, value) def _create_field(self, project_id, field_name, field_type): project_fields = self._target.getProjectCustomFields(project_id) if field_name.lower() not in [f.name.lower() for f in project_fields]: all_fields = self._target.getCustomFields() if field_name.lower() not in [f.name.lower() for f in all_fields]: self._target.createCustomFieldDetailed(field_name, field_type, False, True, False, {}) if field_type in ('string', 'text', 'date', 'integer', 'float', 'period'): self._target.createProjectCustomFieldDetailed( project_id, field_name, 'No ' + field_name) else: bundle_name = field_name + ' bundle' create_bundle_safe(self._target, bundle_name, field_type) self._target.createProjectCustomFieldDetailed( project_id, field_name, 'No ' + field_name, {'bundle': bundle_name}) def _create_field_value(self, project_id, field_name, field_type, value): if field_type.startswith('user'): if hasattr(value, 'name'): value.name = self._create_user(value).login else: value = self._create_user(value).login elif field_type.startswith('version'): value = self._to_yt_version(value) if field_name == 'Assignee': return value if field_name in youtrack.EXISTING_FIELDS: return value self._create_field(project_id, field_name, field_type) if field_type in ('string', 'text', 'date', 'integer', 'float', 'period'): return value field = self._target.getProjectCustomField(project_id, field_name) bundle = self._target.getBundle(field_type, field.bundle) try: if hasattr(value, 'value'): value = value.value elif hasattr(value, 'name') and not (field_type.startswith('version') or field_type.startswith('ownedField')): value = value.name self._target.addValueToBundle(bundle, value) except youtrack.YouTrackException as e: if e.response.status != 409 or e.response.reason.lower( ) != 'conflict': print(e) return value def _get_value_presentation(self, field_type, value): if field_type == 'date': return str(to_unixtime(value)) if field_type == 'integer': return '%d' % int(float(value)) if field_type == 'float': return '%.5f' % float(value) if field_type == 'string': return value if field_type == 'text': return value if field_type == 'period': return '%d' % int(float(value) * 60) if hasattr(value, 'value'): return value.value elif hasattr(value, 'name'): return value.name return value def _create_user(self, user): user = self._to_yt_user(user) if not hasattr(user, 'created'): self._target.createUser(user) user.created = True if hasattr(user, 'groups'): for group in user.groups: self._create_group(group) self._target.setUserGroup(user.login, group.name) return user def _create_group(self, group): if isinstance(group, str): group = self._to_yt_group(group) if not hasattr(group, 'created'): try: self._target.getGroup(group.name) except youtrack.YouTrackException: self._target.createGroup(group) group.created = True return group def _add_journals(self, issue, journals): if not journals: return for rec in journals: if hasattr(rec, 'notes') and rec.notes is not None and rec.notes != '': comment = youtrack.Comment() if self._params.get('use_markdown'): comment.markdown = "true" comment.text = rec.notes comment.author = self._create_user(rec.user).login comment.created = str(to_unixtime(rec.created_on)) issue['comments'].append(comment) def _enable_timetracking(self, project): self._target.setProjectTimeTrackingSettings(project.identifier, enabled=True) def _add_work_items(self, issue): import_data = [] work_items = self._source.get_time_entries(issue.id) flatten_work_items = [] for item in work_items: if 'time_entries' in item.attributes: time_entries = item.time_entries for time_entry in time_entries: flatten_work_items.append(time_entry) else: flatten_work_items.append(item) for t in sorted(flatten_work_items, key=lambda tt: tt.spent_on): work_item = youtrack.WorkItem() work_item.authorLogin = self._create_user(t.user).login work_item.date = str(to_unixtime(t.spent_on)) work_item.description = t.comments work_item.duration = int(float(t.hours) * 60) import_data.append(work_item) if import_data: self._target.importWorkItems(self._get_yt_issue_id(issue), import_data) def _add_tags(self, issue): if not hasattr(issue, 'tags') or not issue.tags: return try: if not self._tags: self._tags = {t.name: t for t in self._target.getTags()} except youtrack.YouTrackException as e: print(str(e)) return try: yt_issue_id = self._get_yt_issue_id(issue) except youtrack.YouTrackException as e: print(str(e)) return assigned_tags = [] for r_tag in issue.tags: try: tag = self._get_or_create_tag(r_tag.name) self._target.executeCommand(yt_issue_id, command='tag ' + tag.name, disable_notifications=True) assigned_tags.append(tag.name) except youtrack.YouTrackException as e: print(str(e)) return print("Issue [ %s ] assigned tags: %s" % (yt_issue_id, ', '.join(assigned_tags))) def _get_or_create_tag(self, tag_name): if tag_name not in self._tags: tag = youtrack.Tag() tag.name = tag_name tag.untagOnResolve = False tag.visibleForGroup = self._get_all_users_group().name tag.updatableByGroup = None self._target.createTag(tag) self._tags[tag_name] = tag return self._tags[tag_name] def _get_all_users_group(self): if not self._all_users_group: self._all_users_group = next( (g for g in self._target.getGroups() if g.name in ['All users', 'Все пользователи']), {'name': None}) return self._all_users_group def _add_attachments(self, issue): if not hasattr(issue, 'attachments'): return max_attempts = 5 for attach in issue.attachments: attach.author.login = self._create_user(attach.author).login if not attach.author.login: attach.author.login = '******' attempts = max_attempts while attempts: attempts -= 1 try: self._target.createAttachmentFromAttachment( self._get_yt_issue_id(issue), RedmineAttachment(attach, self._source)) break except Exception as e: print(e) if attempts: delay = 10 + (max_attempts - attempts - 1) * 10 print(("Can't import attachment: %s. Retry in %d s." % (attach.filename, delay))) time.sleep(delay) else: print(('Failed to import attachment: %s. Skipped.' % attach.filename)) def _collect_relations(self, issue): link_types = { 'duplicates': 'duplicate', 'relates': 'relates', 'blocks': 'depend', 'precedes': 'depend' } if hasattr(issue, 'relations'): for rel in issue.relations: if rel.relation_type not in link_types: print(('Unsuitable link type: %s. Skipped' % rel.relation_type)) continue from_id = rel.issue_id to_id = rel.issue_to_id if rel.relation_type == 'duplicates': from_id, to_id = to_id, from_id self._push_relation(from_id, to_id, link_types[rel.relation_type]) if hasattr(issue, 'children'): for child in issue.children: self._push_relation(issue.id, child.id, 'subtask') def _push_relation(self, from_iid, to_iid, relation_type): from_iid = int(from_iid) to_iid = int(to_iid) if relation_type not in self._relations: self._relations[relation_type] = {} relations = self._relations[relation_type] if from_iid not in relations: relations[from_iid] = {} relations[from_iid][to_iid] = None def _apply_relations(self, limit=DEFAULT_BATCH_SIZE): links = [] for link_type, ids in list(self._relations.items()): for from_id, to_ids in list(ids.items()): for to_id in to_ids: link = youtrack.Link() link.typeName = link_type try: link.source = self._to_yt_issue_id(from_id) link.target = self._to_yt_issue_id(to_id) except KeyError: print(("Cannot apply link (%s) to issues: %d and %d" % (link_type, from_id, to_id))) if hasattr(link, 'source') and link.source: print("The second issue was not imported") continue else: print("The first issues was not imported") break links.append(link) if len(links) >= limit: self._target.importLinks(links) del links[0:] if links: self._target.importLinks(links)
def fb2youtrack(target_url, target_login, target_password, source_url, source_login, source_password, project_names, max_issue_id) : #encoding = 'utf-8' source = FBClient(source_url, source_login, source_password) target = Connection(target_url, target_login, target_password) accessible_projects = source.list_project_names() for p_name in project_names : if not(p_name in accessible_projects.keys()) : print 'Unknown project names. Exiting...' sys.exit() # for p_name in accessible_projects : # if (p_name.encode('utf-8') in project_names_str) : # project_names_str.remove(p_name.encode('utf-8')) # project_names.append(p_name) # # if (len(project_names_str) != 0) : # print 'Unknown project names!' print('Creating custom fields') # # for field_name in ['Category', 'Priority', 'Status']: # field_name = get_yt_name_from_fb__field_name(field_name) # create_custom_field(target, fbugz.CF_TYPES[field_name], field_name, False) fb_category_bundle_name = u'FB Categories' fb_priorities_bundle_name = u'FB Priorities' fb_statuses_bundle_name = u'FB Statuses' common_fields = { u'category' : fb_category_bundle_name, u'priority' : fb_priorities_bundle_name, u'status' : fb_statuses_bundle_name } field_name = u'category' create_bundle_with_values(target,fbugz.CF_TYPES[get_yt_name_from_fb_field_name(field_name)], common_fields[field_name], source.list_categories(), lambda bundle, value : bundle.createElement(to_yt_field_value(field_name, value))) field_name = u'priority' create_bundle_with_values(target, fbugz.CF_TYPES[get_yt_name_from_fb_field_name(field_name)], common_fields[field_name], [elem[0] + '-' + elem[1] for elem in source.list_priorities()], lambda bundle, value : bundle.createElement(to_yt_field_value(field_name, value))) field_name = u'status' statuses = [(to_yt_field_value(field_name, value), resolved) for (value, resolved) in source.list_statuses()] create_bundle_with_values(target, fbugz.CF_TYPES[get_yt_name_from_fb_field_name(field_name)], common_fields[field_name], statuses, lambda bundle, value : to_yt_status(bundle, value)) simple_fields = [u'original_title', u'version', u'computer', u'due', u'estimate'] for name in simple_fields: name = get_yt_name_from_fb_field_name(name) create_custom_field(target, fbugz.CF_TYPES[name], name, False) print 'Importing users' for name in ['Normal', 'Deleted', 'Community', 'Virtual'] : group = Group() group.name = name try : target.createGroup(group) print 'Group with name [ %s ] successfully created' % name except: print "Can't create group with name [ %s ] (maybe because it already exists)" % name users_to_import = [] max = 100 for user in source.get_users() : yt_user = _to_yt_user(user) print 'Importing user [ %s ]' % yt_user.login users_to_import.append(yt_user) if len(users_to_import) >= max: _do_import_users(target, users_to_import) users_to_import = [] _do_import_users(target, users_to_import) print 'Importing users finished' # to handle linked issues try : target.createIssueLinkTypeDetailed('parent-child', 'child of', 'parent of', True) except YouTrackException: print "Can't create issue link type [ parent-child ] (maybe because it already exists)" links_to_import = [] for project_name in project_names : value_sets = dict([]) project_id = accessible_projects[project_name] print 'Importing project [ %s ]' % project_name target.createProjectDetailed(project_id, project_name.encode('utf-8'), 'no description', 'root') print 'Creating custom fields in project [ %s ]' % project_name for cf_name in common_fields: bundle_name = common_fields[cf_name] cf_name = get_yt_name_from_fb_field_name(cf_name) target.deleteProjectCustomField(project_id, cf_name) target.createProjectCustomFieldDetailed(project_id, cf_name, 'No ' + cf_name.lower(), {'bundle' : bundle_name}) for cf_name in simple_fields: cf_name = get_yt_name_from_fb_field_name(cf_name) try: target.createProjectCustomFieldDetailed(project_id, cf_name, 'No ' + cf_name.lower()) except YouTrackException: print "Can't create custom field with name [%s]" % cf_name cf_name = get_yt_name_from_fb_field_name('fix_for') milestones = source.get_milestones(project_id) value_sets["fix_for"] = [] for milestone in milestones: value_sets["fix_for"].append(milestone.name) milestone.name = to_yt_field_value('fix_for', milestone.name) add_values_to_field(target, cf_name, project_id, milestones, lambda bundle, value: _to_yt_version(bundle, value)) cf_name = get_yt_name_from_fb_field_name('area') areas = source.get_areas(project_id) value_sets["area"] = [] for area in areas: value_sets["area"].append(area.name) area.name = to_yt_field_value('area', area.name) add_values_to_field(target, cf_name, project_id, areas, lambda bundle, value: _to_yt_subsystem(bundle, value)) print 'Importing issues for project [ %s ]' % project_name start = 0 issues_to_import = [] # create dictionary with child : parent pairs while start <= max_issue_id: fb_issues = source.get_issues(project_name, start, 30) for issue in fb_issues : add_values_to_field(target, get_yt_name_from_fb_field_name('area'), project_id, [issue.field_values['area']], lambda bundle, value: bundle.createElement(value)) issues_to_import.append(_to_yt_issue(issue, value_sets)) target.importIssues(project_id, project_name.encode('utf-8') + " assignees", issues_to_import) for issue in fb_issues : full_issue_id = '%s-%s' % (project_id, issue.ix_bug) for attach in issue.attachments : target.createAttachmentFromAttachment(full_issue_id, attach) for tag in issue.tags : target.executeCommand(full_issue_id, 'tag ' + tag) if issue.bug_parent is not None: parent_issue_id = '%s-%s' % (source.get_issue_project_id(issue.bug_parent), issue.bug_parent) link = Link() link.typeName = 'parent-child' link.source = full_issue_id link.target = parent_issue_id links_to_import.append(link) issues_to_import = [] start += 30 print 'Importing issues for project [ %s ] finished' % project_name print 'Importing issue links' print target.importLinks(links_to_import) print 'Importing issue links finished'
def fb2youtrack(params): # Connection to FogBugz source = FBClient(params['fb_url'], params['fb_login'], params['fb_password']) # Connecting to YouTrack token = params.get('token') if not token and 'token_file' in params: try: with open(params['token_file'], 'r') as f: token = f.read().strip() except (OSError, IOError) as e: print("Cannot load token from file: " + str(e)) sys.exit(1) if token: target = Connection(params['yt_url'], token=token) elif 'yt_login' in params: target = Connection(params['yt_url'], params.get('yt_login'), params.get('yt_password')) else: print("You have to provide token or login/password to import data") sys.exit(1) if not params.get('project_lead_login'): project_lead = params.get('yt_login') if not project_lead: for login in ('root', 'admin', 'administrator', 'guest'): try: project_lead = target.getUser(login).login break except youtrack.YouTrackException: continue params['project_lead_login'] = project_lead max_issue_id = params['fb_max_issue_id'] project_names = youtrackutils.fbugz.PROJECTS_TO_IMPORT accessible_projects = source.list_project_names() for p_name in project_names: if not (p_name in accessible_projects.keys()): print('Unknown project names. Exiting...') sys.exit() # for p_name in accessible_projects : # if (p_name.encode('utf-8') in project_names_str) : # project_names_str.remove(p_name.encode('utf-8')) # project_names.append(p_name) # # if (len(project_names_str) != 0) : # print 'Unknown project names!' print('Creating custom fields') # # for field_name in ['Category', 'Priority', 'Status']: # field_name = get_yt_name_from_fb__field_name(field_name) # create_custom_field(target, fbugz.CF_TYPES[field_name], field_name, False) fb_category_bundle_name = u'FB Categories' fb_priorities_bundle_name = u'FB Priorities' fb_statuses_bundle_name = u'FB Statuses' common_fields = { u'category' : fb_category_bundle_name, u'priority' : fb_priorities_bundle_name, u'status' : fb_statuses_bundle_name } field_name = u'category' create_bundle_with_values( target, youtrackutils.fbugz.CF_TYPES[get_yt_field_name(field_name)], common_fields[field_name], source.list_categories(), lambda bundle, value: bundle.createElement(to_yt_field_value(field_name, value))) field_name = u'priority' create_bundle_with_values(target, youtrackutils.fbugz.CF_TYPES[get_yt_field_name(field_name)], common_fields[field_name], [elem[0] + '-' + elem[1] for elem in source.list_priorities()], lambda bundle, value : bundle.createElement(to_yt_field_value(field_name, value))) field_name = u'status' statuses = [(to_yt_field_value(field_name, value), resolved) for (value, resolved) in source.list_statuses()] create_bundle_with_values(target, youtrackutils.fbugz.CF_TYPES[get_yt_field_name(field_name)], common_fields[field_name], statuses, lambda bundle, value : to_yt_status(bundle, value)) simple_fields = [u'original_title', u'version', u'computer', u'due', u'estimate'] for name in simple_fields: name = get_yt_field_name(name) create_custom_field(target, youtrackutils.fbugz.CF_TYPES[name], name, False) print('Importing users') for name in ['Normal', 'Deleted', 'Community', 'Virtual'] : group = Group() group.name = name try : target.createGroup(group) print('Group with name [ %s ] successfully created' % name) except: print("Can't create group with name [ %s ] (maybe because it already exists)" % name) users_to_import = [] max = 100 for user in source.get_users() : yt_user = _to_yt_user(user) print('Importing user [ %s ]' % yt_user.login) users_to_import.append(yt_user) if len(users_to_import) >= max: _do_import_users(target, users_to_import) users_to_import = [] _do_import_users(target, users_to_import) print('Importing users finished') # to handle linked issues try : target.createIssueLinkTypeDetailed('parent-child', 'child of', 'parent of', True) except YouTrackException: print("Can't create issue link type [ parent-child ] (maybe because it already exists)") links_to_import = [] for project_name in project_names: value_sets = dict([]) project_id = accessible_projects[project_name] print('Importing project [ %s ]' % project_name) target.createProjectDetailed(project_id, project_name.encode('utf-8'), '', params['project_lead_login']) print('Creating custom fields in project [ %s ]' % project_name) for cf_name in common_fields: bundle_name = common_fields[cf_name] cf_name = get_yt_field_name(cf_name) target.deleteProjectCustomField(project_id, cf_name) target.createProjectCustomFieldDetailed(project_id, cf_name, 'No ' + cf_name.lower(), {'bundle' : bundle_name}) for cf_name in simple_fields: cf_name = get_yt_field_name(cf_name) try: target.createProjectCustomFieldDetailed(project_id, cf_name, 'No ' + cf_name.lower()) except YouTrackException: print("Can't create custom field with name [%s]" % cf_name) cf_name = get_yt_field_name('fix_for') milestones = source.get_milestones(project_id) value_sets["fix_for"] = [] for milestone in milestones: value_sets["fix_for"].append(milestone.name) milestone.name = to_yt_field_value('fix_for', milestone.name) add_values_to_field(target, cf_name, project_id, milestones, lambda bundle, value: _to_yt_version(bundle, value)) cf_name = get_yt_field_name('area') areas = source.get_areas(project_id) value_sets["area"] = [] for area in areas: value_sets["area"].append(area.name) area.name = to_yt_field_value('area', area.name) add_values_to_field(target, cf_name, project_id, areas, lambda bundle, value: _to_yt_subsystem(bundle, value)) print('Importing issues for project [ %s ]' % project_name) start = 0 issues_to_import = [] # create dictionary with child : parent pairs while start <= max_issue_id: fb_issues = source.get_issues(project_name, start, 30) for issue in fb_issues : add_values_to_field(target, get_yt_field_name('area'), project_id, [issue.field_values['area']], lambda bundle, value: bundle.createElement(value)) issues_to_import.append(_to_yt_issue(issue, value_sets)) target.importIssues(project_id, project_name.encode('utf-8') + " assignees", issues_to_import) for issue in fb_issues : full_issue_id = '%s-%s' % (project_id, issue.ix_bug) for attach in issue.attachments : target.createAttachmentFromAttachment(full_issue_id, attach) for tag in issue.tags : target.executeCommand(full_issue_id, 'tag ' + tag) if issue.bug_parent is not None: parent_issue_id = '%s-%s' % (source.get_issue_project_id(issue.bug_parent), issue.bug_parent) link = Link() link.typeName = 'parent-child' link.source = full_issue_id link.target = parent_issue_id links_to_import.append(link) issues_to_import = [] start += 30 print('Importing issues for project [ %s ] finished' % project_name) print('Importing issue links') print(target.importLinks(links_to_import)) print('Importing issue links finished')
continue if params.get('add_new_comments'): target_comments = dict() max_id = 0 for c in target_issue.getComments(): target_comments[c.created] = c if max_id < c.created: max_id = c.created for c in issue.getComments(): if c.created > max_id or c.created not in target_comments: group = None if hasattr(c, 'permittedGroup'): group = c.permittedGroup try: target.executeCommand(issue.id, 'comment', c.text, group, c.author) except youtrack.YouTrackException, e: print 'Cannot add comment to issue ' print e if params.get('sync_custom_fields'): skip_fields = [] if tt_settings and tt_settings.Enabled and tt_settings.TimeSpentField: skip_fields.append(tt_settings.TimeSpentField) skip_fields = [name.lower() for name in skip_fields] for pcf in [pcf for pcf in project_custom_fields if pcf.name.lower() not in skip_fields]: target_cf_value = None if pcf.name in target_issue: target_cf_value = target_issue[pcf.name] if isinstance(target_cf_value, (list, tuple)): target_cf_value = set(target_cf_value)
def fb2youtrack(target_url, target_login, target_password, source_url, source_login, source_password, project_names, max_issue_id): #encoding = 'utf-8' source = FBClient(source_url, source_login, source_password) target = Connection(target_url, target_login, target_password) accessible_projects = source.list_project_names() for p_name in project_names: if not (p_name in accessible_projects.keys()): print 'Unknown project names. Exiting...' sys.exit() # for p_name in accessible_projects : # if (p_name.encode('utf-8') in project_names_str) : # project_names_str.remove(p_name.encode('utf-8')) # project_names.append(p_name) # # if (len(project_names_str) != 0) : # print 'Unknown project names!' print('Creating custom fields') # # for field_name in ['Category', 'Priority', 'Status']: # field_name = get_yt_name_from_fb__field_name(field_name) # create_custom_field(target, fbugz.CF_TYPES[field_name], field_name, False) fb_category_bundle_name = u'FB Categories' fb_priorities_bundle_name = u'FB Priorities' fb_statuses_bundle_name = u'FB Statuses' common_fields = { u'category': fb_category_bundle_name, u'priority': fb_priorities_bundle_name, u'status': fb_statuses_bundle_name } field_name = u'category' create_bundle_with_values( target, fbugz.CF_TYPES[get_yt_name_from_fb_field_name(field_name)], common_fields[field_name], source.list_categories(), lambda bundle, value: bundle.createElement( to_yt_field_value(field_name, value))) field_name = u'priority' create_bundle_with_values( target, fbugz.CF_TYPES[get_yt_name_from_fb_field_name(field_name)], common_fields[field_name], [elem[0] + '-' + elem[1] for elem in source.list_priorities()], lambda bundle, value: bundle.createElement( to_yt_field_value(field_name, value))) field_name = u'status' statuses = [(to_yt_field_value(field_name, value), resolved) for (value, resolved) in source.list_statuses()] create_bundle_with_values( target, fbugz.CF_TYPES[get_yt_name_from_fb_field_name(field_name)], common_fields[field_name], statuses, lambda bundle, value: to_yt_status(bundle, value)) simple_fields = [ u'original_title', u'version', u'computer', u'due', u'estimate' ] for name in simple_fields: name = get_yt_name_from_fb_field_name(name) create_custom_field(target, fbugz.CF_TYPES[name], name, False) print 'Importing users' for name in ['Normal', 'Deleted', 'Community', 'Virtual']: group = Group() group.name = name try: target.createGroup(group) print 'Group with name [ %s ] successfully created' % name except: print "Can't create group with name [ %s ] (maybe because it already exists)" % name users_to_import = [] max = 100 for user in source.get_users(): yt_user = _to_yt_user(user) print 'Importing user [ %s ]' % yt_user.login users_to_import.append(yt_user) if len(users_to_import) >= max: _do_import_users(target, users_to_import) users_to_import = [] _do_import_users(target, users_to_import) print 'Importing users finished' # to handle linked issues try: target.createIssueLinkTypeDetailed('parent-child', 'child of', 'parent of', True) except YouTrackException: print "Can't create issue link type [ parent-child ] (maybe because it already exists)" links_to_import = [] for project_name in project_names: value_sets = dict([]) project_id = accessible_projects[project_name] print 'Importing project [ %s ]' % project_name target.createProjectDetailed(project_id, project_name.encode('utf-8'), 'no description', 'root') print 'Creating custom fields in project [ %s ]' % project_name for cf_name in common_fields: bundle_name = common_fields[cf_name] cf_name = get_yt_name_from_fb_field_name(cf_name) target.deleteProjectCustomField(project_id, cf_name) target.createProjectCustomFieldDetailed(project_id, cf_name, 'No ' + cf_name.lower(), {'bundle': bundle_name}) for cf_name in simple_fields: cf_name = get_yt_name_from_fb_field_name(cf_name) try: target.createProjectCustomFieldDetailed( project_id, cf_name, 'No ' + cf_name.lower()) except YouTrackException: print "Can't create custom field with name [%s]" % cf_name cf_name = get_yt_name_from_fb_field_name('fix_for') milestones = source.get_milestones(project_id) value_sets["fix_for"] = [] for milestone in milestones: value_sets["fix_for"].append(milestone.name) milestone.name = to_yt_field_value('fix_for', milestone.name) add_values_to_field( target, cf_name, project_id, milestones, lambda bundle, value: _to_yt_version(bundle, value)) cf_name = get_yt_name_from_fb_field_name('area') areas = source.get_areas(project_id) value_sets["area"] = [] for area in areas: value_sets["area"].append(area.name) area.name = to_yt_field_value('area', area.name) add_values_to_field( target, cf_name, project_id, areas, lambda bundle, value: _to_yt_subsystem(bundle, value)) print 'Importing issues for project [ %s ]' % project_name start = 0 issues_to_import = [] # create dictionary with child : parent pairs while start <= max_issue_id: fb_issues = source.get_issues(project_name, start, 30) for issue in fb_issues: add_values_to_field( target, get_yt_name_from_fb_field_name('area'), project_id, [issue.field_values['area']], lambda bundle, value: bundle.createElement(value)) issues_to_import.append(_to_yt_issue(issue, value_sets)) target.importIssues(project_id, project_name.encode('utf-8') + " assignees", issues_to_import) for issue in fb_issues: full_issue_id = '%s-%s' % (project_id, issue.ix_bug) for attach in issue.attachments: target.createAttachmentFromAttachment( full_issue_id, attach) for tag in issue.tags: target.executeCommand(full_issue_id, 'tag ' + tag) if issue.bug_parent is not None: parent_issue_id = '%s-%s' % (source.get_issue_project_id( issue.bug_parent), issue.bug_parent) link = Link() link.typeName = 'parent-child' link.source = full_issue_id link.target = parent_issue_id links_to_import.append(link) issues_to_import = [] start += 30 print 'Importing issues for project [ %s ] finished' % project_name print 'Importing issue links' print target.importLinks(links_to_import) print 'Importing issue links finished'
def trac2youtrack(target_url, target_login, target_password, project_ID, project_name, env_path): # creating connection to trac to import issues to client = Client(env_path) # creating connection to youtrack to import issues in target = Connection(target_url, target_login, target_password) #create project print "Creating project[%s]" % project_name try : target.getProject(project_ID) except youtrack.YouTrackException: target.createProjectDetailed(project_ID, project_name, client.get_project_description(), target_login) #importing users trac_users = client.get_users() print "Importing users" yt_users = list([]) # converting trac users to yt users registered_users = set([]) for user in trac_users : print "Processing user [ %s ]" % user.name registered_users.add(user.name) yt_users.append(to_youtrack_user(user)) # adding users to yt project target.importUsers(yt_users) print "Importing users finished" print "Creating project custom fields" create_yt_custom_field(target, project_ID, "Priority", client.get_issue_priorities()) create_yt_custom_field(target, project_ID, "Type", client.get_issue_types()) trac_resolution_to_yt_state = lambda track_field, yt_bundle : to_youtrack_state(track_field, yt_bundle) create_yt_bundle_custom_field(target, project_ID, "Resolution", client.get_issue_resolutions(), trac_resolution_to_yt_state) trac_versions = client.get_versions() trac_version_to_yt_version = lambda trac_field, yt_bundle : to_youtrack_version(trac_field, yt_bundle) create_yt_bundle_custom_field(target, project_ID, "Version", trac_versions, trac_version_to_yt_version) #create_yt_bundle_custom_field(target, project_ID, "Affected versions", trac_versions, trac_version_to_yt_version) trac_components = client.get_components() for cmp in trac_components : if cmp.owner not in registered_users : cmp.owner, registered_users = process_non_authorised_user(target, registered_users, cmp.owner) trac_component_to_yt_subsystem = lambda trac_field, yt_bundle : to_youtrack_subsystem(trac_field, yt_bundle) create_yt_bundle_custom_field(target, project_ID, "Component", trac_components, trac_component_to_yt_subsystem) create_yt_custom_field(target, project_ID, "Severity", client.get_severities()) trac_custom_fields = client.get_custom_fields_declared() check_box_fields = dict([]) for elem in trac_custom_fields: print "Processing custom field [ %s ]" % elem.name type_name = None if elem.type == "checkbox": if len(elem.label) > 0: opt = elem.label else: opt = elem.name options = list([opt]) check_box_fields[elem.name] = opt else: options = elem.options values = None if len(options): values = options field_name = elem.name if field_name in tracLib.FIELD_NAMES.keys() : field_name = tracLib.FIELD_NAMES[field_name] field_type = tracLib.CUSTOM_FIELD_TYPES[elem.type] if field_name in tracLib.FIELD_TYPES.keys(): field_type = tracLib.FIELD_TYPES[field_name] process_custom_field(target, project_ID, field_type, field_name, trac_values_to_youtrack_values(field_name, values)) print "Creating project custom fields finished" print "Importing issues" trac_issues = client.get_issues() yt_issues = list([]) counter = 0 max = 100 for issue in trac_issues: print "Processing issue [ %s ]" % (str(issue.id)) counter += 1 if not (issue.reporter in registered_users): yt_user, registered_users = process_non_authorised_user(target, registered_users, issue.reporter) if yt_user is None : issue.reporter = "guest" else: issue.reporter = yt_user if not (issue.owner in registered_users): yt_user, registered_users = process_non_authorised_user(target, registered_users, issue.owner) if yt_user is None : issue.owner = "" else: issue.owner = yt_user legal_cc = set([]) for cc in issue.cc: if cc in registered_users: legal_cc.add(cc) issue.cc = legal_cc yt_issues.append(to_youtrack_issue(issue, check_box_fields)) if counter == max: counter = 0 print target.importIssues(project_ID, project_name + ' Assignees', yt_issues) yt_issues = list([]) print target.importIssues(project_ID, project_name + ' Assignees', yt_issues) print 'Importing issues finished' #importing tags print "Importing keywords" for issue in trac_issues: print "Importing tags from issue [ %s ]" % (str(issue.id)) tags = issue.keywords for t in tags: target.executeCommand(str(project_ID) + "-" + str(issue.id), "tag " + t.encode('utf-8')) print "Importing keywords finished" print "Importing attachments" for issue in trac_issues: print "Processing issue [ %s ]" % (str(issue.id)) issue_attach = issue.attachment for attach in issue_attach: print "Processing attachment [ %s ]" % attach.filename.encode('utf-8') if not (attach.author_name in registered_users): yt_user, registered_users = process_non_authorised_user(target, registered_users, attach.author_name) if yt_user is None : attach.author_name = "guest" else: attach.author_name = yt_user content = open(urllib.quote(attach.filename.encode('utf-8'))) target.createAttachment(str(project_ID) + "-" + str(issue.id), attach.name, content, attach.author_name, created=attach.time) print "Importing attachments finished" print "Importing workitems" tt_enabled = False for issue in trac_issues: if issue.workitems: if not tt_enabled: tt_settings = target.getProjectTimeTrackingSettings(str(project_ID)) if not tt_settings.Enabled: print "Enabling TimeTracking for the prject" target.setProjectTimeTrackingSettings(str(project_ID), enabled=True) tt_enabled = True print "Processing issue [ %s ]" % (str(issue.id)) workitems = [to_youtrack_workitem(w) for w in issue.workitems] target.importWorkItems(str(project_ID) + "-" + str(issue.id), workitems) print "Importing workitems finished"
link_importer.addAvailableIssues(issues) for issue in issues: try: target_issue = target.getIssue(issue.id) except youtrack.YouTrackException, e: print "Cannot get target issue" print e continue if params.get('sync_tags') and issue.tags: try: for tag in issue.tags: tag = re.sub(r'[,&<>]', '_', tag) try: target.executeCommand(issue.id, 'tag ' + tag, disable_notifications=True) except youtrack.YouTrackException: tag = re.sub(r'[\s-]', '_', tag) target.executeCommand(issue.id, 'tag ' + tag, disable_notifications=True) except youtrack.YouTrackException, e: print "Cannot sync tags for issue " + issue.id print e if params.get('add_new_comments'): target_comments = dict() max_id = 0 for c in target_issue.getComments(): target_comments[c.created] = c if max_id < c.created: max_id = c.created for c in issue.getComments():
def bugzilla2youtrack(target_url, target_login, target_pass, bz_db, bz_host, bz_port, bz_login, bz_pass, bz_product_names, issues_filter): # connecting to bz client = Client(bz_host, int(bz_port), bz_login, bz_pass, db_name=bz_db) if not len(bz_product_names): answer = raw_input("All projects will be imported. Are you sure? [y/n]") if answer.capitalize() != "Y": sys.exit() bz_product_names = client.get_product_names() print "bz_product_names : " + repr(bz_product_names) # connecting to yt target = Connection(target_url, target_login, target_pass) print "Creating issue link types" link_types = client.get_issue_link_types() for link in link_types: print "Processing link type [ %s ]" % link.name try: target.createIssueLinkType(to_yt_issue_link_type(link)) except YouTrackException: print "Can't create link type [ %s ] (maybe because it already exists)" % link.name print "Creating issue link types finished" print "Creating custom fields" custom_fields = client.get_custom_fields() for cf in custom_fields: create_yt_custom_field(cf, target) print "Creating custom fields finished" for key in bugzilla.FIELD_TYPES: if key not in youtrack.EXISTING_FIELDS: create_custom_field(target, bugzilla.FIELD_TYPES[key], key, True, bundle_policy="1") bz_product_ids = [] for name in bz_product_names: product_id = str(client.get_product_id_by_name(name)) bz_product_ids.append(product_id) print "Creating project [ %s ] with name [ %s ]" % (product_id, name) try: target.getProject(str(product_id)) except YouTrackException: target.createProjectDetailed(str(product_id), name, client.get_project_description(product_id), target_login) print "Importing components for project [ %s ]" % product_id process_components(client.get_components(product_id), product_id, target) print "Importing components finished for project [ %s ]" % product_id print "Importing versions for project [ %s ]" % product_id process_versions(client.get_versions(product_id), product_id, target) print "Importing versions finished for project [ %s ] finished" % product_id print "Importing issues to project [ %s ]" % product_id max_count = 100 count = 0 from_id = 0 bz_issues_count = client.get_issues_count(product_id) while count < bz_issues_count: batch = client.get_issues(product_id, from_id, from_id + max_count) batch = [bz_issue for bz_issue in batch if (issues_filter(bz_issue))] count += len(batch) from_id += max_count target.importIssues(product_id, product_id + " assignees", [to_yt_issue(bz_issue, product_id, target) for bz_issue in batch]) # todo convert to good tags import for issue in batch: tags = issue["keywords"] | issue["flags"] for t in tags: print "Processing tag [ %s ]" % t.encode('utf8') target.executeCommand(str(product_id) + "-" + str(issue[get_number_in_project_field_name()]), "tag " + t.encode('utf8')) for issue in batch: for attach in issue["attachments"]: print "Processing attachment [ %s ]" % (attach.name.encode('utf8')) content = StringIO(attach.content) target.createAttachment(str(product_id) + "-" + str(issue[get_number_in_project_field_name()]), attach.name, content, attach.reporter , created=str(int(attach.created) * 1000)) print "Importing issues to project [ %s ] finished" % product_id # todo add pagination to links print "Importing issue links" cf_links = client.get_issue_links() duplicate_links = client.get_duplicate_links() if len(duplicate_links): try: target.createIssueLinkTypeDetailed("Duplicate", "duplicates", "is duplicated by", True) except YouTrackException: print "Can't create link type [ Duplicate ] (maybe because it already exists)" depend_links = client.get_dependencies_link() if len(depend_links): try: target.createIssueLinkTypeDetailed("Depend", "depends on", "is required for", True) except YouTrackException: print "Can't create link type [ Depend ] (maybe because it already exists)" links = cf_links | duplicate_links | depend_links links_to_import = list([]) for link in links: print "Processing link %s for issue%s" % (link.name, link.source) if (str(link.target_product_id) in bz_product_ids) and (str(link.source_product_id) in bz_product_ids): links_to_import.append(to_yt_issue_link(link)) print target.importLinks(links_to_import) print "Importing issue links finished"
def Main(sendTo, subject, yamlMessage): """ Workflow Zabbix-YouTrack :param sendTo: URL to Youtrack (ex. https://youtrack.example.com) :param subject: subject from Zabbix Action :param yamlMessage: message from Zabbix Action :return: """ # ----- Use below example yamlMessage to debug ----- # yamlMessage = """Name: 'Test Zabbix-YT workflow, ignore it' # Text: 'Agent ping (server:agent.ping()): DOWN (1) ' # Hostname: 'server.exmpale.ru' # Status: "OK" # Severity: "High" # EventID: "96976" # TriggerID: "123456789012" """ messages = yaml.load(yamlMessage) # ----- START Issue parameters ----- # Correspondence between the YouTrackPriority and ZabbixSeverity # Critical >= High # Normal < High ytPriority = 'Normal' if messages['Severity'] == 'Disaster' or messages['Severity'] == 'High': ytPriority = 'Critical' ytName = "{} ZabbixTriggerID::{}".format(messages['Name'], messages['TriggerID']) # ----- END Issue parameters ----- # ----- START Youtrack Issue description ----- # Search link to other issue searchString = "Hostname: '{}'".format(messages['Hostname']) linkToHostIssue = "{youtrack}/issues/{projectname}?q={query}".format( youtrack=sendTo, projectname=YT_PROJECT_NAME, query=urllib.parse.quote(searchString, safe='')) issueDescription = """ {ytName} ----- {yamlMessage} ----- - [https://zabbix.example.com/zabbix.php?action=dashboard.view Zabbix Dashboard] - Show [{linkToHostIssue} all issue for *this host*] """.format( ytName=ytName, yamlMessage=yamlMessage, linkToHostIssue=linkToHostIssue, ) # ----- END Youtrack Issue description ----- # ----- START Youtrack current week ----- # Create connect to Youtrack API connection = Connection(sendTo, YT_USER, YT_PASSWORD) # Get current week in YT format (Sprint planned) version = connection.getAllBundles('version') for fixVersion in version[0].values: if fixVersion['archived'] == False and fixVersion['released'] == False: fixVersionWeek = fixVersion['name'] break # ----- END Youtrack current week ----- # ----- START Youtrack get or create issue ----- # Get issue if exist # Search for TriggerID createNewIssue = False logger.debug("Get issue with text '{}'".format(messages['TriggerID'])) issue = connection.getIssues( YT_PROJECT_NAME, "ZabbixTriggerID::{}".format(messages['TriggerID']), 0, 1) if len(issue) == 0: createNewIssue = True else: # if issue contains TriggerID in summary, then it's good issue # else create new issue, this is bad issue, not from Zabbix if "ZabbixTriggerID::{}".format( messages['TriggerID']) in issue[0]['summary']: issueId = issue[0]['id'] issue = connection.getIssue(issueId) else: createNewIssue = True # Create new issue if createNewIssue: logger.debug("Create new issue because it is not exist") issue = connection.createIssue( YT_PROJECT_NAME, 'Unassigned', ytName, issueDescription, priority=ytPriority, subsystem=YT_SUBSYSTEM, type=YT_TYPE, ) time.sleep(3) # Parse ID for new issue result = re.search(r'(CM-\d*)', issue[0]['location']) issueId = result.group(0) issue = connection.getIssue(issueId) logger.debug("Issue have id={}".format(issueId)) # Set issue service ExecAndLog(connection, issueId, "Service {}".format(YT_SERVICE)) # Update priority ExecAndLog(connection, issueId, "Priority {}".format(ytPriority)) # ----- END Youtrack get or create issue ----- # ----- START PROBLEM block ------ if messages['Status'] == "PROBLEM": # Issue exist and NOT Hold on, Unnassigned and Estimated time set if issue['State'] != 'Hold on': # Estimated time ExecAndLog(connection, issueId, "Estimated time {}".format(YT_TIME)) # Update fix version ExecAndLog(connection=connection, issueId=issueId, command="Sprint planned {}".format(fixVersionWeek)) # Reopen if Fixed or Verified or Canceled if issue['State'] == 'Fixed' or issue['State'] == 'Verified' or issue[ 'State'] == 'Canceled': # Reopen Issue ExecAndLog(connection, issueId, "State reopen") # Assignee issue ExecAndLog(connection, issueId, command="Assignee Unassigned") # Update summary and description for issue logger.debug("Run command in {issueId}: {command}".format( issueId=issueId, command= """Update summary and description with connection.updateIssue method""" )) connection.updateIssue(issueId=issueId, summary=ytName, description=issueDescription) # Add comment logger.debug("Run command in {issueId}: {command}".format( issueId=issueId, command="""Now is PROBLEM {}""".format(messages['Text']))) connection.executeCommand(issueId=issueId, command="", comment=YT_COMMENT.format( status=messages['Status'], text=messages['Text'])) # Send ID to Zabbix: logger.debug("ZABBIX-API: Send Youtrack ID to {}".format( messages['EventID'])) Zbx.event.acknowledge(eventids=messages['EventID'], message="Create Youtrack task") Zbx.event.acknowledge( eventids=messages['EventID'], message="https://youtrack.example.com/issue/{}".format(issueId)) # ----- End PROBLEM block ------ # ----- Start OK block ----- if messages['Status'] == "OK": if issue['State'] == 'Hold on' or issue['State'] == 'Registered': # Cancel if not in work ExecAndLog(connection, issueId, command="State Cancel") # Assignee issue ExecAndLog(connection, issueId, command="Assignee {}".format(YT_ASSIGNEE)) if issue['State'] == 'Fixed': # Verify if Fixed ExecAndLog(connection, issueId, command="State verify") logger.debug("Run command in {issueId}: {command}".format( issueId=issueId, command="""Now is OK {}""".format(messages['Text']))) connection.executeCommand(issueId=issueId, command="", comment=YT_COMMENT.format( status=messages['Status'], text=messages['Text']))
def youtrack2youtrack(source_url, source_login, source_password, target_url, target_login, target_password, project_ids, query='', source_token=None, target_token=None, params=None): if not len(project_ids): print("You should sign at least one project to import") return if params is None: params = {} source = Connection(source_url, source_login, source_password) if (source_token is None) else Connection(source_url, token=source_token) target = Connection(target_url, target_login, target_password) if (target_token is None) else Connection(target_url, token=target_token) # , proxy_info = httplib2.ProxyInfo(socks.PROXY_TYPE_HTTP, 'localhost', 8888) print("Import issue link types") for ilt in source.getIssueLinkTypes(): try: print(target.createIssueLinkType(ilt)) except youtrack.YouTrackException as e: print(e.message) user_importer = UserImporter(source, target, caching_users=params.get('enable_user_caching', True)) link_importer = LinkImporter(target) # create all projects with minimum info and project lead set created_projects = [] for project_id in project_ids: created = create_project_stub(source, target, project_id, user_importer) created_projects.append(created) # save created project ids to create correct group roles afterwards user_importer.addCreatedProjects([project.id for project in created_projects]) # import project leads with group they are included and roles assigned to these groups user_importer.importUsersRecursively([target.getUser(project.lead) for project in created_projects]) # afterwards in a script any user import imply recursive import cf_names_to_import = set([]) # names of cf prototypes that should be imported for project_id in project_ids: cf_names_to_import.update([pcf.name.capitalize() for pcf in source.getProjectCustomFields(project_id)]) target_cf_names = [pcf.name.capitalize() for pcf in target.getCustomFields()] period_cf_names = [] for cf_name in cf_names_to_import: source_cf = source.getCustomField(cf_name) if source_cf.type.lower() == 'period': period_cf_names.append(source_cf.name.lower()) print("Processing custom field '%s'" % utf8encode(cf_name)) if cf_name in target_cf_names: target_cf = target.getCustomField(cf_name) if not (target_cf.type == source_cf.type): print("In your target and source YT instances you have field with name [ %s ]" % utf8encode(cf_name)) print("They have different types. Source field type [ %s ]. Target field type [ %s ]" % (source_cf.type, target_cf.type)) print("exiting...") exit() else: if hasattr(source_cf, "defaultBundle"): create_bundle_from_bundle(source, target, source_cf.defaultBundle, source_cf.type, user_importer) target.createCustomField(source_cf) failed_commands = [] for projectId in project_ids: source = Connection(source_url, source_login, source_password) if (source_token is None) else Connection( source_url, token=source_token) target = Connection(target_url, target_login, target_password) if (target_token is None) else Connection( target_url, token=target_token) # , proxy_info = httplib2.ProxyInfo(socks.PROXY_TYPE_HTTP, 'localhost', 8888) # reset connections to avoid disconnections user_importer.resetConnections(source, target) link_importer.resetConnections(target) # copy project, subsystems, versions project = source.getProject(projectId) link_importer.addAvailableIssuesFrom(projectId) project_custom_fields = source.getProjectCustomFields(projectId) # create bundles and additional values for pcf_ref in project_custom_fields: pcf = source.getProjectCustomField(projectId, pcf_ref.name) if hasattr(pcf, "bundle"): try: create_bundle_from_bundle(source, target, pcf.bundle, source.getCustomField(pcf.name).type, user_importer) except youtrack.YouTrackException as e: if e.response.status != 409: raise e else: print(e) target_project_fields = [pcf.name.lower() for pcf in target.getProjectCustomFields(projectId)] for field in project_custom_fields: if field.name.lower() in target_project_fields: if hasattr(field, 'bundle'): if field.bundle != target.getProjectCustomField(projectId, field.name).bundle: target.deleteProjectCustomField(projectId, field.name) create_project_custom_field(target, field, projectId) else: try: create_project_custom_field(target, field, projectId) except youtrack.YouTrackException as e: if e.response.status != 409: raise e else: print(e) # copy issues start = 0 max = 20 sync_workitems = enable_time_tracking(source, target, projectId) tt_settings = target.getProjectTimeTrackingSettings(projectId) print("Import issues") last_created_issue_number = 0 while True: try: print("Get issues from " + str(start) + " to " + str(start + max)) issues = source.getIssues(projectId, query, start, max) if len(issues) <= 0: break if convert_period_values and period_cf_names: for issue in issues: for pname in period_cf_names: for fname in issue.__dict__: if fname.lower() != pname: continue issue[fname] = period_to_minutes(issue[fname]) users = set([]) for issue in issues: print("Collect users for issue [%s]" % issue.id) users.add(issue.getReporter()) if issue.hasAssignee(): if isinstance(issue.Assignee, (list, tuple)): users.update(issue.getAssignee()) else: users.add(issue.getAssignee()) # TODO: http://youtrack.jetbrains.net/issue/JT-6100 users.add(issue.getUpdater()) if issue.hasVoters(): users.update(issue.getVoters()) for comment in issue.getComments(): users.add(comment.getAuthor()) print("Collect links for issue [%s]" % issue.id) link_importer.collectLinks(issue.getLinks(True)) # links.extend(issue.getLinks(True)) # fix problem with comment.text for comment in issue.getComments(): if not hasattr(comment, "text") or (len(comment.text.strip()) == 0): setattr(comment, 'text', 'no text') user_importer.importUsersRecursively(users) print("Create issues [" + str(len(issues)) + "]") if params.get('create_new_issues'): create_issues(target, issues, last_created_issue_number) else: print(target.importIssues(projectId, project.name + ' Assignees', issues)) link_importer.addAvailableIssues(issues) for issue in issues: try: target_issue = target.getIssue(issue.id) except youtrack.YouTrackException as e: print("Cannot get target issue") print(e) continue if params.get('sync_tags') and issue.tags: try: for tag in issue.tags: tag = re.sub(r'[,&<>]', '_', tag) try: target.executeCommand(issue.id, 'tag ' + tag, disable_notifications=True) except youtrack.YouTrackException: tag = re.sub(r'[\s-]', '_', tag) target.executeCommand(issue.id, 'tag ' + tag, disable_notifications=True) except youtrack.YouTrackException as e: print("Cannot sync tags for issue " + issue.id) print(e) if params.get('add_new_comments'): target_comments = dict() max_id = 0 for c in target_issue.getComments(): target_comments[c.created] = c if max_id < c.created: max_id = c.created for c in issue.getComments(): if c.created > max_id or c.created not in target_comments: group = None if hasattr(c, 'permittedGroup'): group = c.permittedGroup try: target.executeCommand(issue.id, 'comment', c.text, group, c.author, disable_notifications=True) except youtrack.YouTrackException as e: print('Cannot add comment to issue') print(e) if params.get('sync_custom_fields'): skip_fields = [] if tt_settings and tt_settings.Enabled and tt_settings.TimeSpentField: skip_fields.append(tt_settings.TimeSpentField) skip_fields = [name.lower() for name in skip_fields] for pcf in [pcf for pcf in project_custom_fields if pcf.name.lower() not in skip_fields]: target_cf_value = None if pcf.name in target_issue: target_cf_value = target_issue[pcf.name] if isinstance(target_cf_value, (list, tuple)): target_cf_value = set(target_cf_value) elif target_cf_value == target.getProjectCustomField(projectId, pcf.name).emptyText: target_cf_value = None source_cf_value = None if pcf.name in issue: source_cf_value = issue[pcf.name] if isinstance(source_cf_value, (list, tuple)): source_cf_value = set(source_cf_value) elif source_cf_value == source.getProjectCustomField(projectId, pcf.name).emptyText: source_cf_value = None if source_cf_value == target_cf_value: continue if isinstance(source_cf_value, set) or isinstance(target_cf_value, set): if source_cf_value is None: source_cf_value = set([]) elif not isinstance(source_cf_value, set): source_cf_value = set([source_cf_value]) if target_cf_value is None: target_cf_value = set([]) elif not isinstance(target_cf_value, set): target_cf_value = set([target_cf_value]) for v in target_cf_value: if v not in source_cf_value: target.executeCommand(issue.id, 'remove %s %s' % (pcf.name, v), disable_notifications=True) for v in source_cf_value: if v not in target_cf_value: target.executeCommand(issue.id, 'add %s %s' % (pcf.name, v), disable_notifications=True) else: if source_cf_value is None: source_cf_value = target.getProjectCustomField(projectId, pcf.name).emptyText if pcf.type.lower() == 'date': m = re.match(r'(\d{10})(?:\d{3})?', str(source_cf_value)) if m: source_cf_value = datetime.datetime.fromtimestamp( int(m.group(1))).strftime('%Y-%m-%d') elif pcf.type.lower() == 'period': source_cf_value = '%sm' % source_cf_value command = '%s %s' % (pcf.name, source_cf_value) try: target.executeCommand(issue.id, command, disable_notifications=True) except youtrack.YouTrackException as e: if e.response.status == 412 and e.response.reason.find('Precondition Failed') > -1: print('WARN: Some workflow blocks following command: %s' % command) failed_commands.append((issue.id, command)) if sync_workitems: workitems = source.getWorkItems(issue.id) if workitems: existing_workitems = dict() target_workitems = target.getWorkItems(issue.id) if target_workitems: for w in target_workitems: _id = '%s\n%s\n%s' % (w.date, w.authorLogin, w.duration) if hasattr(w, 'description'): _id += '\n%s' % w.description existing_workitems[_id] = w new_workitems = [] for w in workitems: _id = '%s\n%s\n%s' % (w.date, w.authorLogin, w.duration) if hasattr(w, 'description'): _id += '\n%s' % w.description if _id not in existing_workitems: new_workitems.append(w) if new_workitems: print("Process workitems for issue [ " + issue.id + "]") try: user_importer.importUsersRecursively( [source.getUser(w.authorLogin) for w in new_workitems]) target.importWorkItems(issue.id, new_workitems) except youtrack.YouTrackException as e: if e.response.status == 404: print("WARN: Target YouTrack doesn't support workitems importing.") print("WARN: Workitems won't be imported.") sync_workitems = False else: print("ERROR: Skipping workitems because of error:" + str(e)) print("Process attachments for issue [%s]" % issue.id) existing_attachments = dict() try: for a in target.getAttachments(issue.id): existing_attachments[a.name + '\n' + a.created] = a except youtrack.YouTrackException as e: if e.response.status == 404: print("Skip importing attachments because issue %s doesn't exist" % issue.id) continue raise e attachments = [] users = set([]) for a in issue.getAttachments(): if a.name + '\n' + a.created in existing_attachments and not params.get('replace_attachments'): a.name = utf8encode(a.name) try: print("Skip attachment '%s' (created: %s) because it's already exists" % (utf8encode(a.name), utf8encode(a.created))) except Exception: pass continue attachments.append(a) author = a.getAuthor() if author is not None: users.add(author) user_importer.importUsersRecursively(users) for a in attachments: print("Transfer attachment of " + utf8encode(issue.id) + ": " + utf8encode(a.name)) # TODO: add authorLogin to workaround http://youtrack.jetbrains.net/issue/JT-6082 # a.authorLogin = target_login try: target.createAttachmentFromAttachment(issue.id, a) except BaseException as e: print("Cant import attachment [ %s ]" % utf8encode(a.name)) print(repr(e)) continue if params.get('replace_attachments'): try: old_attachment = existing_attachments.get(a.name + '\n' + a.created) if old_attachment: print('Deleting old attachment') target.deleteAttachment(issue.id, old_attachment.id) except BaseException as e: print("Cannot delete attachment '%s' from issue %s" % ( utf8encode(a.name), utf8encode(issue.id))) print(e) except Exception as e: print('Cant process issues from ' + str(start) + ' to ' + str(start + max)) traceback.print_exc() raise e start += max print("Import issue links") link_importer.importCollectedLinks() print("Trying to execute failed commands once again") for issue_id, command in failed_commands: try: print('Executing command on issue %s: %s' % (issue_id, command)) target.executeCommand(issue_id, command, disable_notifications=True) except youtrack.YouTrackException as e: print('Failed to execute command for issue #%s: %s' % (issue_id, command)) print(e)
def bugzilla2youtrack(params): # Connecting to Bugzilla client = Client(host=params['bz_host'], port=int(params['bz_port']), login=params['bz_login'], password=params['bz_password'], db_name=params['bz_db']) bz_product_names = params.get('bz_product_names') if not bz_product_names: answer = raw_input( "All projects will be imported. Are you sure? [Y/n] ") if answer.strip().lower() not in ("y", "yes", ""): sys.exit() bz_product_names = client.get_product_names() print("bz_product_names: " + repr(bz_product_names)) # Connecting to YouTrack token = params.get('token') if not token and 'token_file' in params: try: with open(params['token_file'], 'r') as f: token = f.read().strip() except (OSError, IOError) as e: print("Cannot load token from file: " + str(e)) sys.exit(1) if token: target = Connection(params['yt_url'], token=token) elif 'yt_login' in params: target = Connection(params['yt_url'], params.get('yt_login'), params.get('yt_password')) else: print("You have to provide token or login/password to import data") sys.exit(1) print("Creating issue link types") link_types = client.get_issue_link_types() for link in link_types: print("Processing link type [ %s ]" % link.name) try: target.createIssueLinkType(to_yt_issue_link_type(link)) except YouTrackException: print( "Can't create link type [ %s ] (maybe because it already exists)" % link.name) print("Creating issue link types finished") print("Creating custom fields") custom_fields = client.get_custom_fields() for cf in custom_fields: create_yt_custom_field(cf, target) print("Creating custom fields finished") for key in youtrackutils.bugzilla.FIELD_TYPES: if key not in youtrack.EXISTING_FIELDS: create_custom_field(target, youtrackutils.bugzilla.FIELD_TYPES[key], key, True, bundle_policy="1") bz_product_ids = [] for name in bz_product_names: product_id = str(client.get_product_id_by_name(name)) bz_product_ids.append(product_id) print("Creating project [ %s ] with name [ %s ]" % (product_id, name)) try: target.getProject(product_id) except YouTrackException: target.createProjectDetailed( product_id, name, client.get_project_description(product_id), 'root') print("Importing components for project [ %s ]" % product_id) process_components(client.get_components(product_id), product_id, target) print("Importing components finished for project [ %s ]" % product_id) print("Importing versions for project [ %s ]" % product_id) process_versions(client.get_versions(product_id), product_id, target) print("Importing versions finished for project [ %s ] finished" % product_id) print("Importing issues to project [ %s ]" % product_id) max_count = 100 count = 0 from_id = 0 bz_issues_count = client.get_issues_count(product_id) while count < bz_issues_count: batch = client.get_issues(product_id, from_id, from_id + max_count) batch = [bz_issue for bz_issue in batch] count += len(batch) from_id += max_count target.importIssues(product_id, product_id + " assignees", [ to_yt_issue(bz_issue, product_id, target) for bz_issue in batch ]) # todo convert to good tags import for issue in batch: tags = issue["keywords"] | issue["flags"] for t in tags: print("Processing tag [ %s ]" % t.encode('utf8')) target.executeCommand( str(product_id) + "-" + str(issue[get_number_in_project_field_name()]), "tag " + t.encode('utf8')) for issue in batch: issue_id = str(product_id) + '-' + \ str(issue[get_number_in_project_field_name()]) for attach in issue["attachments"]: print("Processing attachment [ %s ] for issue %s" % (utf8encode(attach.name), issue_id)) content = StringIO(attach.content) try: target.importAttachment( issue_id, attach.name, content, attach.reporter.login, None, None, str(int(attach.created) * 1000)) except urllib2.HTTPError as e: print("WARN: Cant import attachment [ %s ]" % utf8encode(attach.name)) print(e.code) print(e.read()) print("Please check Max Upload File Size in YouTrack") continue print("Importing issues to project [ %s ] finished" % product_id) # todo add pagination to links print("Importing issue links") cf_links = client.get_issue_links() duplicate_links = client.get_duplicate_links() if len(duplicate_links): try: target.createIssueLinkTypeDetailed("Duplicate", "duplicates", "is duplicated by", True) except YouTrackException: print( "Can't create link type [ Duplicate ] (maybe because it already exists)" ) depend_links = client.get_dependencies_link() if len(depend_links): try: target.createIssueLinkTypeDetailed("Depend", "depends on", "is required for", True) except YouTrackException: print( "Can't create link type [ Depend ] (maybe because it already exists)" ) links = cf_links | duplicate_links | depend_links links_to_import = list([]) for link in links: print("Processing link %s for issue%s" % (link.name, link.source)) if (str(link.target_product_id) in bz_product_ids) and (str( link.source_product_id) in bz_product_ids): links_to_import.append(to_yt_issue_link(link)) print(target.importLinks(links_to_import)) print("Importing issue links finished")
def bugzilla2youtrack(target_url, target_login, target_pass, bz_db, bz_host, bz_port, bz_login, bz_pass, bz_product_names, issues_filter): # connecting to bz client = Client(bz_host, int(bz_port), bz_login, bz_pass, db_name=bz_db) if not len(bz_product_names): answer = raw_input("All projects will be imported. Are you sure? [y/n]") if answer.capitalize() != "Y": sys.exit() bz_product_names = client.get_product_names() print("bz_product_names : " + repr(bz_product_names)) # connecting to yt target = Connection(target_url, target_login, target_pass) print("Creating issue link types") link_types = client.get_issue_link_types() for link in link_types: print("Processing link type [ %s ]" % link.name) try: target.createIssueLinkType(to_yt_issue_link_type(link)) except YouTrackException: print("Can't create link type [ %s ] (maybe because it already exists)" % link.name) print("Creating issue link types finished") print("Creating custom fields") custom_fields = client.get_custom_fields() for cf in custom_fields: create_yt_custom_field(cf, target) print("Creating custom fields finished") for key in youtrackutils.bugzilla.FIELD_TYPES: if key not in youtrack.EXISTING_FIELDS: create_custom_field(target, youtrackutils.bugzilla.FIELD_TYPES[key], key, True, bundle_policy="1") bz_product_ids = [] for name in bz_product_names: product_id = str(client.get_product_id_by_name(name)) bz_product_ids.append(product_id) print("Creating project [ %s ] with name [ %s ]" % (product_id, name)) try: target.getProject(str(product_id)) except YouTrackException: target.createProjectDetailed(str(product_id), name, client.get_project_description(product_id), target_login) print("Importing components for project [ %s ]" % product_id) process_components(client.get_components(product_id), product_id, target) print("Importing components finished for project [ %s ]" % product_id) print("Importing versions for project [ %s ]" % product_id) process_versions(client.get_versions(product_id), product_id, target) print("Importing versions finished for project [ %s ] finished" % product_id) print("Importing issues to project [ %s ]" % product_id) max_count = 100 count = 0 from_id = 0 bz_issues_count = client.get_issues_count(product_id) while count < bz_issues_count: batch = client.get_issues(product_id, from_id, from_id + max_count) batch = [bz_issue for bz_issue in batch if (issues_filter(bz_issue))] count += len(batch) from_id += max_count target.importIssues(product_id, product_id + " assignees", [to_yt_issue(bz_issue, product_id, target) for bz_issue in batch]) # todo convert to good tags import for issue in batch: tags = issue["keywords"] | issue["flags"] for t in tags: print("Processing tag [ %s ]" % t.encode('utf8')) target.executeCommand(str(product_id) + "-" + str(issue[get_number_in_project_field_name()]), "tag " + t.encode('utf8')) for issue in batch: for attach in issue["attachments"]: print("Processing attachment [ %s ]" % (attach.name.encode('utf8'))) content = StringIO(attach.content) target.createAttachment(str(product_id) + "-" + str(issue[get_number_in_project_field_name()]), attach.name, content, attach.reporter.login , created=str(int(attach.created) * 1000)) print("Importing issues to project [ %s ] finished" % product_id) # todo add pagination to links print("Importing issue links") cf_links = client.get_issue_links() duplicate_links = client.get_duplicate_links() if len(duplicate_links): try: target.createIssueLinkTypeDetailed("Duplicate", "duplicates", "is duplicated by", True) except YouTrackException: print("Can't create link type [ Duplicate ] (maybe because it already exists)") depend_links = client.get_dependencies_link() if len(depend_links): try: target.createIssueLinkTypeDetailed("Depend", "depends on", "is required for", True) except YouTrackException: print("Can't create link type [ Depend ] (maybe because it already exists)") links = cf_links | duplicate_links | depend_links links_to_import = list([]) for link in links: print("Processing link %s for issue%s" % (link.name, link.source)) if (str(link.target_product_id) in bz_product_ids) and (str(link.source_product_id) in bz_product_ids): links_to_import.append(to_yt_issue_link(link)) print(target.importLinks(links_to_import)) print("Importing issue links finished")
def Main(sendTo, subject, yamlMessage): """ Workflow Zabbix-YouTrack :param sendTo: URL to Youtrack (ex. https://youtrack.example.com) :param subject: subject from Zabbix Action :param yamlMessage: message from Zabbix Action :return: """ # ----- Use below example yamlMessage to debug ----- # yamlMessage = """Name: 'Test Zabbix-YT workflow, ignore it' # Text: 'Agent ping (server:agent.ping()): DOWN (1) ' # Hostname: 'server.exmpale.ru' # Status: "OK" # Severity: "High" # EventID: "96976" # TriggerID: "123456789012" """ messages = yaml.load(yamlMessage) # ----- START Issue parameters ----- # Correspondence between the YouTrackPriority and ZabbixSeverity # Critical >= High # Normal < High ytPriority = 'Normal' if messages['Severity'] == 'Disaster' or messages['Severity'] == 'High': ytPriority = 'Critical' ytName = "{} ZabbixTriggerID::{}".format(messages['Name'], messages['TriggerID']) # ----- END Issue parameters ----- # ----- START Youtrack Issue description ----- # Search link to other issue searchString = "Hostname: '{}'".format(messages['Hostname']) linkToHostIssue = "{youtrack}/issues/{projectname}?q={query}".format( youtrack=sendTo, projectname=YT_PROJECT_NAME, query=urllib.parse.quote(searchString, safe='') ) issueDescription = """ {ytName} ----- {yamlMessage} ----- - [https://zabbix.example.com/zabbix.php?action=dashboard.view Zabbix Dashboard] - Show [{linkToHostIssue} all issue for *this host*] """.format( ytName=ytName, yamlMessage=yamlMessage, linkToHostIssue=linkToHostIssue, ) # ----- END Youtrack Issue description ----- # ----- START Youtrack current week ----- # Create connect to Youtrack API connection = Connection(sendTo, YT_USER, YT_PASSWORD) # Get current week in YT format (Sprint planned) version = connection.getAllBundles('version') for fixVersion in version[0].values: if fixVersion['archived'] == False and fixVersion['released'] == False: fixVersionWeek = fixVersion['name'] break # ----- END Youtrack current week ----- # ----- START Youtrack get or create issue ----- # Get issue if exist # Search for TriggerID createNewIssue = False logger.debug("Get issue with text '{}'".format(messages['TriggerID'])) issue = connection.getIssues(YT_PROJECT_NAME, "ZabbixTriggerID::{}".format(messages['TriggerID']), 0, 1) if len(issue) == 0: createNewIssue = True else: # if issue contains TriggerID in summary, then it's good issue # else create new issue, this is bad issue, not from Zabbix if "ZabbixTriggerID::{}".format(messages['TriggerID']) in issue[0]['summary']: issueId = issue[0]['id'] issue = connection.getIssue(issueId) else: createNewIssue = True # Create new issue if createNewIssue: logger.debug("Create new issue because it is not exist") issue = connection.createIssue(YT_PROJECT_NAME, 'Unassigned', ytName, issueDescription, priority=ytPriority, subsystem=YT_SUBSYSTEM, type=YT_TYPE, ) time.sleep(3) # Parse ID for new issue result = re.search(r'(CM-\d*)', issue[0]['location']) issueId = result.group(0) issue = connection.getIssue(issueId) logger.debug("Issue have id={}".format(issueId)) # Set issue service ExecAndLog(connection, issueId, "Service {}".format(YT_SERVICE)) # Update priority ExecAndLog(connection, issueId, "Priority {}".format(ytPriority)) # ----- END Youtrack get or create issue ----- # ----- START PROBLEM block ------ if messages['Status'] == "PROBLEM": # Issue exist and NOT Hold on, Unnassigned and Estimated time set if issue['State'] != 'Hold on': # Estimated time ExecAndLog(connection, issueId, "Estimated time {}".format(YT_TIME)) # Update fix version ExecAndLog(connection=connection, issueId=issueId, command="Sprint planned {}".format(fixVersionWeek)) # Reopen if Fixed or Verified or Canceled if issue['State'] == 'Fixed' or issue['State'] == 'Verified' or issue['State'] == 'Canceled': # Reopen Issue ExecAndLog(connection, issueId, "State reopen") # Assignee issue ExecAndLog(connection, issueId, command="Assignee Unassigned") # Update summary and description for issue logger.debug("Run command in {issueId}: {command}".format(issueId=issueId, command="""Update summary and description with connection.updateIssue method""" )) connection.updateIssue(issueId=issueId, summary=ytName, description=issueDescription) # Add comment logger.debug("Run command in {issueId}: {command}".format(issueId=issueId, command="""Now is PROBLEM {}""".format( messages['Text']) )) connection.executeCommand(issueId=issueId, command="", comment=YT_COMMENT.format( status=messages['Status'], text=messages['Text']) ) # Send ID to Zabbix: logger.debug("ZABBIX-API: Send Youtrack ID to {}".format(messages['EventID'])) Zbx.event.acknowledge(eventids=messages['EventID'], message="Create Youtrack task") Zbx.event.acknowledge(eventids=messages['EventID'], message="https://youtrack.example.com/issue/{}".format(issueId)) # ----- End PROBLEM block ------ # ----- Start OK block ----- if messages['Status'] == "OK": if issue['State'] == 'Hold on' or issue['State'] == 'Registered': # Cancel if not in work ExecAndLog(connection, issueId, command="State Cancel") # Assignee issue ExecAndLog(connection, issueId, command="Assignee {}".format(YT_ASSIGNEE)) if issue['State'] == 'Fixed': # Verify if Fixed ExecAndLog(connection, issueId, command="State verify") logger.debug("Run command in {issueId}: {command}".format(issueId=issueId, command="""Now is OK {}""".format(messages['Text']) )) connection.executeCommand(issueId=issueId, command="", comment=YT_COMMENT.format( status=messages['Status'], text=messages['Text']) )