#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tkinter as tk
from tkinter import messagebox, filedialog
from urllib.parse import urlparse
import threading
from prettytable import PrettyTable
import json
import webbrowser
import pyperclip
import pprint
import time
import os
if not os.path.exists("my_link_list.json"): d = {} with open("my_link_list.json", "w") as f: json.dump(d, f)
#print("Created file 'my_link_list.json'")
class Collector:
def __init__(self):
self.root = tk.Tk()
self.root.title("Link Collector") self.root.geometry("575x623") self.root.configure(bg="gray88")
self.root.resizable(height=tk.FALSE,width=tk.FALSE)
currentDir = tk.StringVar()
currentDir.set(os.getcwd())
self.my_url = tk.StringVar()
self.selMode = 'normal'
self.search = tk.StringVar()
self.num_result = 0
self.num_links = 0
with open("my_link_list.json") as f: self.link_list = json.load(f)
#print(self.link_list)
tk.Entry(self.root,textvariable=currentDir,width=95).place(x=0,y=0)
self.urlEntry = tk.Entry(self.root,textvariable=self.my_url,width=43,font=("arial",18)) self.urlEntry.place(x=5,y=35)
tk.Button(self.root,text="SAVE URL AS:",width=49,bg="gray77",command=self.enter_name).place(x=5,y=70)#79
tk.Button(self.root,text="CLEAR URL",width=28,bg="gray77",command=self.clear_url).place(x=363,y=70)
self.canvas = tk.Canvas(self.root,bg="black")
self.canvas.place(x=5,y=110)
self.scrollbar = tk.Scrollbar(self.canvas,orient=tk.VERTICAL)
self.scrollbar2 = tk.Scrollbar(self.canvas,orient=tk.HORIZONTAL)
self.scrollbar.pack(side=tk.RIGHT,fill=tk.Y)
self.scrollbar2.pack(side=tk.BOTTOM,fill=tk.X)
self.linkBox = tk.Listbox(self.canvas,height=30,width=55)# 32 55
self.linkBox.pack()
self.linkBox.config(yscrollcommand = self.scrollbar.set)
self.linkBox.config(xscrollcommand = self.scrollbar2.set)
self.scrollbar.config(command = self.linkBox.yview)
self.scrollbar2.config(command = self.linkBox.xview)
self.searchEntry = tk.Entry(self.root,textvariable=self.search,font=("arial",14),width=13) self.searchEntry.place(x=363,y=110)
tk.Button(self.root,text="SEARCH",bg="gray77",command=self.search_name).place(x=513,y=110)
self.showAll = tk.Button(self.root,text="SHOW ALL LINKS",width=28,command=self.show_all)
self.showAll.place(x=363,y=138)
self.numLinks = tk.Label(self.root,text='{} LINKS'.format(len(self.link_list)),bg='black',fg='green',width=25,font=("arial",10)) self.numLinks.place(x=363,y=195)#205
tk.Button(self.root,text="ADD NEW LINK",bg="gray77",width=28,height=2,command=self.init_copy).place(x=363,y=230)#245
tk.Button(self.root,text="ACCESS",bg="gray77",width=28,height=2,command=self.init_task).place(x=363,y=280)#295
tk.Button(self.root,text="DELETE",bg="gray77",width=28,height=2,command=self.remove_link).place(x=363,y=350)#365
tk.Button(self.root,text="DELETE ALL",bg="gray77",width=28,height=2,command=self.delete_listbox).place(x=363,y=400)#415
tk.Button(self.root,text="CLEAR SELECTION",bg="gray77",width=28,height=2,command=self.clear_selection).place(x=363,y=450)#465
self.selMod = tk.Button(self.root,text="SELECTION MODE: NORMAL",bg="gray77",width=28,height=2,command=self.selection_mode)
self.selMod.place(x=363,y=500)#515
tk.Button(self.root,text="SAVE LIST",bg="gray77",width=28,height=2,command=self.write_doc).place(x=363,y=570)#575
self.show_list()
self.root.mainloop()
def copy_paste(self):
messagebox.showinfo("COPY URL","Copy the URL you want.") self.ultima_copia = pyperclip.paste().strip()
while True:
time.sleep(0.1)
self.copia = pyperclip.paste().strip()
if self.copia != self.ultima_copia:
self.my_url.set(self.copia)
self.ultima_copia = self.copia
break
def clear_url(self):
self.my_url.set("")
def show_all(self):
self.linkBox.delete(0,tk.END)
with open("my_link_list.json") as f: self.link_list = json.load(f)
self.show_list()
#self.showAll.configure(state='disabled')
def delete_listbox(self):
if self.linkBox.size() > 0:
message = messagebox.askquestion("REMOVING",'Do you want to remove all link list?') if message == "yes":
self.link_list = {} self.update_json()
self.linkBox.delete(0,tk.END)
self.numLinks.configure(text='{} LINKS'.format(len(self.link_list)))#rep
def selection_mode(self):
if self.selMode == 'normal':
self.linkBox.configure(selectmode='multiple')
self.selMod.configure(text="SELECTION MODE: MULTIPLE")
self.selMode = 'multiple'
else:
self.linkBox.configure(selectmode='normal')
self.selMod.configure(text="SELECTION MODE: NORMAL")
self.selMode = 'normal'
def enter_name(self):
if self.urlEntry.get() != "":
if self.urlEntry.get() not in self.link_list.values():
is_url = self.validate_url(self.urlEntry.get())
if is_url:
self.window = tk.Tk()######
self.window.geometry("470x300") self.window.title("Link Name") tk.Label(self.window,text="ENTER LINK NAME",width=66).place(x=1,y=45)
entry_name = tk.Entry(self.window,width=25,font=('arial',20)) entry_name.place(x=44,y=90)
entry_name.focus()
tk.Button(self.window,text="SET NAME",width=10,height=2,bg="gray77",command=lambda:self.set_name(entry_name.get())).place(x=193,y=180)
else:
messagebox.showwarning("Invalid URL","Enter a valid URL.") self.my_url.set("") else:
value_name = self.get_key(self.urlEntry.get())
messagebox.showwarning("ALREADY SAVED","The URL provided is already saved as \'{}\'".format(value_name)) else:
messagebox.showwarning("NO URL","No URL provided.")
def search_name(self):
c = 1
self.my_list = []
self.linkBox.delete(0,tk.END)
for i in self.link_list.keys():
if self.searchEntry.get().lower() in i.lower():
self.linkBox.insert(tk.END,(str(c)+"- "+i))
self.my_list.append(self.link_list[i])
c+=1
self.num_result = len(self.my_list)
self.numLinks.configure(text='{} LINKS'.format(self.num_result)) self.showAll.configure(state='normal')
def show_list(self):
if len(self.link_list) > 0:
self.my_list = []
c = 1
for i in self.link_list:
self.linkBox.insert(tk.END,(str(c)+"- "+i))
self.my_list.append(self.link_list[i])
c+=1
self.num_links = len(self.my_list)
self.numLinks.configure(text='{} LINKS'.format(self.num_links)) self.showAll.configure(state='disabled')
def set_name(self,entry_name):
if entry_name != "":
self.window.destroy()
self.linkBox.delete(0,tk.END)
self.link_list[entry_name] = self.urlEntry.get()
self.update_json()
self.show_list()
self.numLinks.configure(text='{} LINKS'.format(len(self.link_list)))
def update_json(self):
with open("my_link_list.json", "w") as f: json.dump(self.link_list, f)
def remove_link(self):
any_selected = self.is_any_selected()
if any_selected:
message = messagebox.askquestion("REMOVE LINK",'Delete selected link/s from link list?') if message == "yes":
for i in self.linkBox.curselection():
del self.link_list[self.get_key(self.my_list[i])]
self.linkBox.delete(0,tk.END)
self.update_json()
if self.showAll["state"]=='disabled':########################
self.show_list()
else:
self.search_name()########################
self.numLinks.configure(text='{} LINKS'.format(self.num_links))
def get_key(self,val):
for key, value in self.link_list.items():
if val == value:
return key
def open_page(self):
try:
for i in self.linkBox.curselection():
print(i)
webbrowser.open_new(self.my_list[i])
except Exception as e:
messagebox.showwarning("Access trouble", str(e))
def clear_selection(self):
for i in self.linkBox.curselection():
self.linkBox.selection_clear(i)
def is_any_selected(self):
for i in range(0,self.linkBox.size()):
if self.linkBox.selection_includes(i):
sel = True
break
else:
sel = False
return sel
def init_task(self):
self.any_selected = self.is_any_selected()
if self.any_selected:
t = threading.Thread(target=self.open_page)
t.start()
else:
messagebox.showwarning("No Link Selected","Select a link to go.")
def create_list_of_lists(self,dic):
listas = []
for k, v in dic.items():
lista = []
lista.append(k)
lista.append(v)
listas.append(lista)
return listas
def write_doc(self):
if self.linkBox.size() > 0:
doc = filedialog.asksaveasfilename(initialdir="/",
title="Save as", initialfile="saved links",defaultextension=".txt")
if doc != "":
new_file = open(doc,"w")
x = PrettyTable()
x.field_names = ["Link Name", "URL"]
content = self.create_list_of_lists(self.link_list)
x.add_rows(content)
new_file.write(x.get_string())
new_file.close()
messagebox.showinfo("SAVED","File saved successfully") else:
messagebox.showwarning("NO ITEMS","There's nothing to save.")
def init_copy(self):
tc = threading.Thread(target=self.copy_paste)
tc.start()
def get_key(self,val):
for key, value in self.link_list.items():
if val == value:
return key
def validate_url(self,url):
try:
result = urlparse(url)
return all([result.scheme, result.netloc])
except ValueError:
return False
if __name__=="__main__":
Collector()
