107 lines
3.6 KiB
Python
107 lines
3.6 KiB
Python
# From: https://www.geeksforgeeks.org/how-to-bind-the-enter-key-to-a-t//kinter-window/
|
|
|
|
# importing tkinter
|
|
|
|
import tkinter as tk
|
|
|
|
from generic import (domain_lookup, ip_lookup, sanitize, spf_dmarc,
|
|
type_hosts_listed, type_hosts_organized)
|
|
|
|
|
|
# function to run on pressing
|
|
# 'ENTER' key from keyboard.
|
|
def key_handler_function(event):
|
|
inputs = entry.get()
|
|
print(type(inputs))
|
|
sanitized = sanitize.strip_and_list(inputs)
|
|
typed = type_hosts_listed.type_hosts(sanitized)
|
|
textbox.insert("end", typed)
|
|
listed = type_hosts_organized.type_hosts(sanitized)
|
|
# IPs = listed['IPs']
|
|
# below works for updating a global variable (here)...
|
|
var.set(listed["IPs"])
|
|
ip_listbox.delete(0, "end")
|
|
domain_listbox.delete(0, "end")
|
|
url_listbox.delete(0, "end")
|
|
email_listbox.delete(0, "end")
|
|
|
|
for IP in listed["IPs"]:
|
|
hostname = ip_lookup.get_hostname(IP)
|
|
output = "{} => {}".format(IP, hostname[0])
|
|
ip_listbox.insert("end", output)
|
|
for domain in listed["domains"]:
|
|
ip = domain_lookup.get_ip_address(domain)
|
|
output = "{} => {}".format(domain, ip)
|
|
domain_listbox.insert("end", output)
|
|
for URL in listed["URLs"]:
|
|
url_listbox.insert("end", URL)
|
|
for email in listed["emails"]:
|
|
dmarc, spf = spf_dmarc.lookup(email)
|
|
output_dmarc = "{} => {}".format(email, dmarc)
|
|
output_spf = "{} => {}".format(email, spf)
|
|
email_listbox.insert("end", output_dmarc, output_spf)
|
|
|
|
|
|
def test_button_function():
|
|
# testing quoting and comma separating hosts, IPs for now, from global StringVar var
|
|
textbox.delete(1.0, "end")
|
|
IPs = ", ".join([f'"{host}"' for host in var.get()])
|
|
textbox.insert("end", IPs)
|
|
print("var is : {}".format(var.get()))
|
|
# ... and printing an updated global variable here!
|
|
|
|
|
|
# main application window
|
|
root = tk.Tk()
|
|
root.title("Enter IP addresses, URLs, domains, email addresses")
|
|
root.geometry("900x600")
|
|
root.resizable(0, 0)
|
|
|
|
# configure the grid
|
|
root.columnconfigure(0, weight=1)
|
|
root.columnconfigure(1, weight=1)
|
|
root.columnconfigure(2, weight=1)
|
|
root.columnconfigure(3, weight=1)
|
|
|
|
# Listing stuff
|
|
|
|
# creating widgets
|
|
entry = tk.Entry(root, text="Put in IP addresses, URLs, domains, email addresses")
|
|
textbox = tk.Text(root, height=8)
|
|
# textbox.insert('1.0', test)
|
|
entrylabel = tk.Label(root, text="Copy-paste or type and Enter :D")
|
|
ip_label = tk.Label(root, text="IP and hostname")
|
|
ip_listbox = tk.Listbox(root)
|
|
domain_label = tk.Label(root, text="domain => IP")
|
|
domain_listbox = tk.Listbox(root)
|
|
url_label = tk.Label(root, text="URL => IP")
|
|
url_listbox = tk.Listbox(root)
|
|
email_label = tk.Label(root, text="email address => DMARC/SPF")
|
|
email_listbox = tk.Listbox(root)
|
|
var = tk.Variable()
|
|
print(var)
|
|
entrylabel.grid(column=1, row=0, columnspan=2, sticky="nsew")
|
|
entry.grid(column=0, row=1, columnspan=4, sticky="nsew")
|
|
ip_label.grid(column=0, row=2, columnspan=2, sticky="nsew")
|
|
ip_listbox.grid(column=0, row=3, columnspan=2, sticky="nsew")
|
|
domain_label.grid(column=2, row=2, columnspan=2, sticky="nsew")
|
|
domain_listbox.grid(column=2, row=3, columnspan=2, sticky="nsew")
|
|
url_label.grid(column=0, row=4, columnspan=2, sticky="nsew")
|
|
url_listbox.grid(column=0, row=5, columnspan=2, sticky="nsew")
|
|
email_label.grid(column=2, row=4, columnspan=2, sticky="nsew")
|
|
email_listbox.grid(column=2, row=5, columnspan=2, sticky="nsew")
|
|
|
|
# label.grid(column=1, row=2)
|
|
# textbox.grid(column=1, row=3)
|
|
|
|
# Trying out buttons
|
|
test_button = tk.Button(
|
|
root, text="comma-separate and add quotes", command=test_button_function
|
|
)
|
|
|
|
# Binding ENTER key to function
|
|
entry.bind("<Return>", key_handler_function)
|
|
|
|
# run the application
|
|
root.mainloop()
|