1
0
Fork 0
forked from fnux/rt2zammad
rt2zammad/migrate.py
2019-05-25 11:04:45 +02:00

96 lines
2.4 KiB
Python
Executable file

#!/usr/bin/env python
"""
Quick and dirty attempt to migrate issues from Request Tracker to Zammad.
"""
import pickle
import json
import os
import sys
from requests.exceptions import HTTPError
from zammad_py import ZammadAPI
from zammad_py.api import TagList
from rt import Rt, ALL_QUEUES
TEMPLATE = """{
"zammad_host": "",
"zammad_user": "",
"zammad_password": "",
"zammad_secure": true,
"rt_url": "",
"rt_user": "",
"rt_pass": ""
}
"""
if not os.path.exists('rt2zammad.json'):
print('Missing rt2zammad.json!')
print('Create one based on following template:')
print(TEMPLATE)
sys.exit(1)
with open('rt2zammad.json') as handle:
config = json.load(handle)
target = ZammadAPI(
host=config['zammad_host'],
username=config['zammad_user'],
password=config['zammad_password'],
is_secure=config['zammad_secure'],
)
target.user.me()
source = Rt(config['rt_url'], config['rt_user'], config['rt_pass'])
if not source.login():
print('Failed to login to RT!')
sys.exit(2)
if os.path.exists('rt2zammad.cache'):
# Load RT from cache
with open('rt2zammad.cache', 'rb') as handle:
data = pickle.load(handle)
users = data['users']
queues = data['queues']
tickets = data['tickets']
else:
# Load RT from remote
users = {}
tickets = []
queues = set()
def ensure_user(username):
if username not in users:
users[username] = source.get_user(username)
for i in range(1, 1000):
print('Loading ticket {}'.format(i))
ticket = source.get_ticket(i)
if ticket is None:
break
queues.add(ticket['Queue'])
ensure_user(ticket['Creator'])
ensure_user(ticket['Owner'])
history = source.get_history(i)
attachments = []
for a in source.get_attachments_ids(i):
attachment = source.get_attachment(i, a)
attachments.append(attachments)
ensure_user(attachment['Creator'])
tickets.append({
'ticket': ticket,
'history': history,
'attachments': attachments,
})
with open('rt2zammad.cache', 'wb') as handle:
data = pickle.dump({'users': users, 'queues': queues, 'tickets': tickets}, handle)
# Create tags
tag_list = TagList(target)
tags = {tag['name'] for tag in tag_list.all()}
for queue in queues:
queue = queue.lower().split()[0]
if queue not in tags:
tag_list.create({'name': queue})