Port Scanner Code Snippet 11

save_results_code.py

import socket
import threading
from datetime import datetime
import tkinter as tk
from tkinter import ttk, font, scrolledtext
from tkinter.filedialog import asksaveasfilename
from ctypes import windll


class App(tk.Tk):
    def __init__(self):
        ...

        # Save Results Button
        self.save_button = ttk.Button(self, text="Save Results", command=self.save_results)
        self.save_button.grid(column=1, row=9, columnspan=3, sticky=tk.EW)
        self.save_button.config(state='disabled')
        
        ...

    # Write results in file
    def write_file(self, file): ...

    # Save results in file
    def save_results(self):
        file = asksaveasfilename(defaultextension=".txt",
                                 initialfile=f'Scan_Results_{datetime.now().strftime("%Y%m%d-%H%M%S")}.txt')
        if file:
            self.write_file(file)


if __name__ == "__main__":
    app = App()
    windll.shcore.SetProcessDpiAwareness(1)
    app.mainloop()

write_file_code.py

import socket
import threading
from datetime import datetime
import tkinter as tk
from tkinter import ttk, font, scrolledtext
from ctypes import windll


class App(tk.Tk):
    def __init__(self):
      ...

    # Write results in file
    def write_file(self, file):
        with open(file, 'w') as f:
            f.write("Port scanner results : \n")
            f.write("-"*22 + "\n")
            f.write(f" Target : \t {self.target} \n")
            if self.var.get() != 3:
                f.write(f" Ports : \t {str(self.ports)} \n")
            else:
                f.write(f" Ports : \t "
                        f"[{str(self.port_range_entry_1.get())} - {str(self.port_range_entry_2.get())}]\n")
            f.write(f"\n Results : \t {str(len(self.results))}/{str(len(self.ports))} \n")
            f.write(f"{str(self.results_area.get('1.0', tk.END))}")

    ...
    

if __name__ == "__main__":
    app = App()
    windll.shcore.SetProcessDpiAwareness(1)
    app.mainloop()