diff --git a/fs2zammad b/fs2zammad index 5a90119..442df8c 100755 --- a/fs2zammad +++ b/fs2zammad @@ -58,19 +58,26 @@ def get_zammad_session(impersonated_user=None): def maybe_create_zammad_user(userdata, zammad_session, attr="login", default=None): # Map disabled users (in RT) to mock user in Zammad. + #print("*userdata",userdata) #debug + #print("*USERMAP", USERMAP) #debug if type(userdata) is str and userdata.lower() in USERMAP: email = userdata + #print("*1") #debug elif type(userdata) is str or "EmailAddress" not in userdata: userdata = { 'EmailAddress': 'technik@ungleich.ch', 'RealName': 'Disabled RT' } email = userdata["EmailAddress"] + #print("*2") #debug elif "EmailAddress" in userdata: email = userdata["EmailAddress"] + #print("*3") #debuig else: raise ValueError("Invalid userdata") + #print("*default",default) #debug + #print("*email",email) #debug # We manually filter out invalid addresses. if email == "*@*.com": userdata = { @@ -81,6 +88,7 @@ def maybe_create_zammad_user(userdata, zammad_session, attr="login", default=Non lowercase_email = email.lower() + #print("*lowercase_email",lowercase_email) #debug if lowercase_email not in USERMAP: kwargs = {"email": email} @@ -107,7 +115,8 @@ def maybe_create_zammad_user(userdata, zammad_session, attr="login", default=Non USERMAP[lowercase_email] = user else: print(f"Could not create/fetch user {lowercase_email}") - + + #print("*lastdefault",default) #debug if default is None: return USERMAP[lowercase_email][attr] @@ -170,7 +179,13 @@ def create_zammad_ticket(id, zammad, h2t, retries=3): print(f"Created Zammad ticket {zammad_ticket['id']} for {label}") if rt_ticket["ticket"]["Owner"] and rt_ticket["ticket"]["Owner"] != "Nobody": - zammad_owner_id = maybe_create_zammad_user(rt_ticket["ticket"]["Owner"], zammad, "id") + #print("*rt_ticket",rt_ticket) #debug + get_owner = rt_ticket["ticket"]["Owner"] + with open(f"users/{get_owner}", "rb") as handle: + get_owner_data = pickle.load(handle) + #print("*get_owner_data",get_owner_data) #debug + zammad_owner_id = maybe_create_zammad_user(get_owner_data, zammad, "id") + #print("*zammad_owner_id",zammad_owner_id) #debug zammad.ticket.update( zammad_ticket["id"], {"owner_id": zammad_owner_id} ) @@ -220,7 +235,9 @@ def create_zammad_ticket(id, zammad, h2t, retries=3): with open(f"users/{entry_creator}", "rb") as handle: rt_entry_creator = pickle.load(handle) + #print("*rt_entry_creator",rt_entry_creator) #debug zammad_entry_creator = maybe_create_zammad_user(rt_entry_creator, zammad) + #print("*zammad_entry_creator",zammad_entry_creator) # debug entry_content = entry["Content"] if entry_content == '' and may_contain_entry_content: file = may_contain_entry_content @@ -236,20 +253,29 @@ def create_zammad_ticket(id, zammad, h2t, retries=3): "internal": entry["Type"] == "Comment", "attachments": files, } - entry_creator_id = maybe_create_zammad_user(zammad_entry_creator, zammad, "id") + entry_creator_id = maybe_create_zammad_user(rt_entry_creator, zammad, "id") # We temporarly change the ticket's creator to create a new entry # without giving its author 'Agent' privileges. restore_creator_to = None if entry_creator_id != zammad_ticket["customer_id"]: + #print("*t1_1 restore_creator_to",restore_creator_to) #debug + #print("*t1_2 entry_creator_id",entry_creator_id) #debug + #print("*t1_3 customer_id",zammad_ticket["customer_id"]) #debug zammad.ticket.update(zammad_ticket["id"], {"customer_id": entry_creator_id}) restore_creator_to = zammad_ticket["customer_id"] zammad_ticket_id = zammad_ticket["id"] print(f"-> Adding a new entry/comment to ticket {zammad_ticket_id} ({zammad_entry_creator})...") + #print("*t2_1",restore_creator_to) #debug + #print("*t2_2",entry_creator_id) #debug + #print("*t2_3",zammad_entry_creator) #debug + #print("*t2_4",zammad_entry_template) #debug TicketArticle(get_zammad_session(zammad_entry_creator)).create(zammad_entry_template) if restore_creator_to != None: + #print("*t3_1",restore_creator_to) #debug + #print("*t3_2",entry_creator_id) #debug zammad.ticket.update(zammad_ticket["id"], {"customer_id": restore_creator_to}) # Returns the new Zammad ticket ID. @@ -295,6 +321,10 @@ else: h2t = html2text.HTML2Text() zammad = get_zammad_session() +#this_page = zammad.user.all() +#for user in this_page: +# print(user) + os.makedirs("users", exist_ok=True) os.makedirs("tickets", exist_ok=True) os.makedirs("attachments", exist_ok=True) @@ -309,7 +339,9 @@ ticket_ids = list(map(int, ticket_ids[import_start-1:import_end])) for id in ticket_ids: try: - matching_zammad_tickets= zammad.ticket.search(f"title: \"\[RT-{id}\]*\"") + print(id) + matching_zammad_tickets = zammad.ticket.search(f"title: \"\[RT-{id}\]*\"") + print(len(matching_zammad_tickets)) if len(matching_zammad_tickets) >= 1: print(f"Found duplicates: {id}") continue diff --git a/zammad-remove-ticket b/zammad-remove-ticket new file mode 100755 index 0000000..4d2f1bc --- /dev/null +++ b/zammad-remove-ticket @@ -0,0 +1,115 @@ +#!/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, +} +""" + + +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_tickets_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) >= 1: + print(f"Found ticket:") + for zt in matching_zammad_tickets: + #print(f"{zt["rt_id"]} {zt["title"]}") + matching_zammad_ids.append(zt["id"]) + + if len(matching_zammad_ids) >= 1: + 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_tickets_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_tickets_for(id, zammad)