forked from fnux/rt2zammad
122 lines
3.2 KiB
Python
Executable file
122 lines
3.2 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
import base64
|
|
import json
|
|
import os
|
|
import pickle
|
|
import sys
|
|
import html2text
|
|
import traceback
|
|
import time
|
|
|
|
from zammad_py import ZammadAPI
|
|
from zammad_py.api import Resource, TagList, TicketArticle
|
|
|
|
TEMPLATE = """{
|
|
"zammad_url": "",
|
|
"zammad_user": "",
|
|
"zammad_password": "",
|
|
"rt_url": "",
|
|
"rt_user": "",
|
|
"rt_pass": "",
|
|
"rt_start": 1,
|
|
"rt_end": 1000,
|
|
}
|
|
"""
|
|
|
|
COMMENT_TEMPLATE = """
|
|
Ticket imported from Request Tracker (RT-#{numerical_id})
|
|
|
|
URL: https://support.ungleich.ch/Ticket/Display.html?id={numerical_id}
|
|
Created: {Created}
|
|
Resolved: {Resolved}
|
|
"""
|
|
|
|
STATUSMAP = {"new": 1, "open": 2, "resolved": 4, "rejected": 4, "deleted": 4}
|
|
USERMAP = {}
|
|
|
|
ZAMMAD_SESSIONS = {}
|
|
|
|
### helpers ###
|
|
|
|
def get_zammad_session(impersonated_user=None):
|
|
if impersonated_user in ZAMMAD_SESSIONS:
|
|
return ZAMMAD_SESSIONS[impersonated_user]
|
|
else:
|
|
kwargs = {}
|
|
if impersonated_user:
|
|
kwargs["on_behalf_of"] = impersonated_user
|
|
session = ZammadAPI(
|
|
url=config["zammad_host"],
|
|
username=config["zammad_user"],
|
|
password=config["zammad_password"],
|
|
**kwargs,
|
|
)
|
|
ZAMMAD_SESSIONS[impersonated_user or config["zammad_user"]] = session
|
|
|
|
return session
|
|
|
|
def remove_duplicates_for(rt_id, zammad, retries=0):
|
|
try:
|
|
matching_zammad_tickets= zammad.ticket.search(f"title: \"\[RT-{rt_id}\]*\"")
|
|
#matching_zammad_tickets= zammad.ticket.search('number:16014')
|
|
#print(matching_zammad_tickets[0])
|
|
matching_zammad_ids = []
|
|
if len(matching_zammad_tickets) >= 2:
|
|
print(f"Found duplicates:")
|
|
for zt in matching_zammad_tickets:
|
|
#print(f"{zt["rt_id"]} {zt["title"]}")
|
|
matching_zammad_ids.append(zt["id"])
|
|
|
|
if len(matching_zammad_ids) >= 2:
|
|
matching_zammad_ids.sort()
|
|
matching_zammad_ids.pop()
|
|
for zt_id in matching_zammad_ids:
|
|
print(f"Deleting Zammad ticket {zt_id}")
|
|
zammad.ticket.destroy(zt_id)
|
|
|
|
except:
|
|
print(f"Failed to process RT-{rt_id} .. ({retries} retries left)")
|
|
if retries > 0:
|
|
print("Sleeping 5 seconds to give Zammad some air...")
|
|
time.sleep(5)
|
|
remove_duplicates_for(id, retries - 1)
|
|
else:
|
|
traceback.print_exc()
|
|
raise RuntimeError
|
|
|
|
### main logic ###
|
|
|
|
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)
|
|
|
|
if len(sys.argv) != 3:
|
|
print(len(sys.argv))
|
|
print("It needs ticket range to import ex. start, end")
|
|
sys.exit()
|
|
else:
|
|
import_start = int(sys.argv[1])
|
|
import_end = int(sys.argv[2])
|
|
print(f"start : {import_start} end : {import_end}")
|
|
|
|
h2t = html2text.HTML2Text()
|
|
zammad = get_zammad_session()
|
|
|
|
os.makedirs("tickets", exist_ok=True)
|
|
|
|
|
|
ticket_ids = os.listdir("tickets/")
|
|
print(f"Found {len(ticket_ids)} tickets on filesystem.")
|
|
ticket_ids = list(map(int, ticket_ids))
|
|
ticket_ids.sort()
|
|
ticket_ids = list(map(int, ticket_ids[import_start-1:import_end]))
|
|
|
|
for id in ticket_ids:
|
|
print(f"Processing RT-{id}...")
|
|
remove_duplicates_for(id, zammad)
|