def __init__(self, audiopaths_and_text, hparams):
self.audiopaths_and_text = load_filepaths_and_text(audiopaths_and_text)
self.text_cleaners = hparams.text_cleaners
self.max_wav_value = hparams.max_wav_value
self.sampling_rate = hparams.sampling_rate
self.load_mel_from_disk = hparams.load_mel_from_disk
self.stft = layers.TacotronSTFT(hparams.filter_length,
hparams.hop_length, hparams.win_length,
hparams.n_mel_channels,
hparams.sampling_rate,
hparams.mel_fmin, hparams.mel_fmax)
self.ct = CustomText(hparams.level)
random.seed(hparams.seed)
random.shuffle(self.audiopaths_and_text)
class TextMelLoader(torch.utils.data.Dataset):
"""
1) loads audio,text pairs
2) normalizes text and converts them to sequences of one-hot vectors
3) computes mel-spectrograms from audio files.
"""
def __init__(self, audiopaths_and_text, hparams):
self.audiopaths_and_text = load_filepaths_and_text(audiopaths_and_text)
self.text_cleaners = hparams.text_cleaners
self.max_wav_value = hparams.max_wav_value
self.sampling_rate = hparams.sampling_rate
self.load_mel_from_disk = hparams.load_mel_from_disk
self.stft = layers.TacotronSTFT(hparams.filter_length,
hparams.hop_length, hparams.win_length,
hparams.n_mel_channels,
hparams.sampling_rate,
hparams.mel_fmin, hparams.mel_fmax)
self.ct = CustomText(hparams.level)
random.seed(hparams.seed)
random.shuffle(self.audiopaths_and_text)
def get_mel_text_pair(self, audiopath_and_text):
# separate filename and text
audiopath, text = audiopath_and_text[0], audiopath_and_text[1]
text = self.get_text(text)
mel = self.get_mel(audiopath)
return (text, mel)
def get_mel(self, filename):
if not self.load_mel_from_disk:
audio, sampling_rate = load_wav_to_torch(filename)
if sampling_rate != self.stft.sampling_rate:
raise ValueError("{} {} SR doesn't match target {} SR".format(
sampling_rate, self.stft.sampling_rate))
audio_norm = audio / self.max_wav_value
audio_norm = audio_norm.unsqueeze(0)
audio_norm = torch.autograd.Variable(audio_norm,
requires_grad=False)
melspec = self.stft.mel_spectrogram(audio_norm)
melspec = torch.squeeze(melspec, 0)
else:
melspec = torch.from_numpy(np.load(filename))
assert melspec.size(0) == self.stft.n_mel_channels, (
'Mel dimension mismatch: given {}, expected {}'.format(
melspec.size(0), self.stft.n_mel_channels))
return melspec
def get_text(self, text):
text_norm = torch.IntTensor(
self.ct.text_to_sequence(text, self.text_cleaners))
return text_norm
def __getitem__(self, index):
return self.get_mel_text_pair(self.audiopaths_and_text[index])
def __len__(self):
return len(self.audiopaths_and_text)
def __init__(self,
filename,
preprocess_config,
train_config,
model_config,
sort=False,
drop_last=False):
self.dataset_name = preprocess_config["dataset"]
self.preprocessed_path = preprocess_config["path"]["preprocessed_path"]
self.cleaners = preprocess_config["preprocessing"]["text"][
"text_cleaners"]
self.batch_size = train_config["optimizer"]["batch_size"]
self.ct = CustomText(model_config['level'])
# getattr(self, 'text_to_sequence', ct.text_to_sequence)
self.basename, self.speaker, self.text, self.raw_text = self.process_meta(
filename)
with open(os.path.join(self.preprocessed_path, "speakers.json")) as f:
self.speaker_map = json.load(f)
self.sort = sort
self.drop_last = drop_last
def build(self):
self.root = Tk()
self.root.geometry("800x622+200+200")
self.root.title("SecureMe")
self.root.resizable(width=FALSE, height=FALSE)
menubar = Menu(self.root)
optionsmenu = Menu(menubar, tearoff=0)
optionsmenu.add_command(label="Refresh (Ctrl+r)",
command=lambda: self.refresh("NONE"))
optionsmenu.add_command(label="Exit", command=lambda: self.quitMenu())
menubar.add_cascade(label="Options", menu=optionsmenu)
self.root.config(menu=menubar)
self.toolbar = ToolBar(self.root, self)
style = Style()
style.configure("BW.TLabel",
foreground="black",
background="slate gray",
borderwidth=2)
mFrame = Frame(self.root, bg="slate gray", padx=25, pady=25)
mFrame.pack(fill=BOTH)
self.notebook = ttk.Notebook(mFrame, height=500, style="BW.TLabel")
primaryFrame = Frame(self.notebook, padx=25, pady=25)
usersFrame = Frame(self.notebook, padx=25, pady=25)
firewallFrame = Frame(self.notebook, padx=25, pady=25)
servicesFrame = Frame(self.notebook, padx=25, pady=25)
processesFrame = Frame(self.notebook, padx=25, pady=25)
self.notebook.add(primaryFrame, text='Primary')
self.notebook.add(usersFrame, text='Users')
self.notebook.add(firewallFrame, text='Firewall')
self.notebook.add(servicesFrame, text='Services')
self.notebook.add(processesFrame, text='Processes')
self.notebook.pack(fill=X)
self.updatebar = Frame(self.root)
self.updatebar.pack(side=BOTTOM, fill=X)
self.left_label = Label(self.updatebar, text="Status: None")
self.left_label.pack(side=LEFT, fill=X)
# Primary Panel
primary_label = Label(primaryFrame,
text='Primary Settings',
font=self.liberation_font_15)
primary_label.grid(row=0, column=0, columnspan=2, sticky=N + S + E + W)
actionspanel = LabelFrame(primaryFrame,
text='System Actions',
padx=10,
pady=10)
actionspanel.grid(row=1, column=0, sticky=E + N, padx=25, pady=25)
openterminal = Button(actionspanel,
text='Open Terminal',
command=lambda: self.openTerminal())
openterminal.pack(padx=5, pady=5)
opencontrol = Button(actionspanel,
text='Open Control Panel',
command=lambda: self.openControlPanel())
opencontrol.pack(padx=5, pady=5)
shutdown = Button(actionspanel,
text='Shutdown',
command=lambda: self.shutdown())
shutdown.pack(padx=5, pady=5)
rebootButton = Button(actionspanel,
text='Reboot',
command=lambda: self.reboot())
rebootButton.pack(padx=5, pady=5)
updatespanel = LabelFrame(primaryFrame,
text='System Updates',
padx=10,
pady=10)
updatespanel.grid(row=1, column=1, sticky=W + N, padx=25, pady=25)
update_button = Button(updatespanel,
text='Basic Update',
command=lambda: self.basicUpdate())
update_button.pack(padx=5, pady=5)
upgrade_button = Button(updatespanel,
text='Basic Upgrade',
command=lambda: self.basicUpgrade())
upgrade_button.pack(padx=5, pady=5)
packageupdate_button = Button(updatespanel,
text='Package Update',
command=lambda: self.packageUpdate())
packageupdate_button.pack(padx=5, pady=5)
# Users Panel
users_label = Label(usersFrame,
text='User Security Settings',
font=self.liberation_font_15)
users_label.pack()
editusers = Frame(usersFrame)
editusers.pack()
addusr = Button(editusers,
text='Add User...',
command=lambda: self.addUser())
addusr.grid(row=0, column=0, padx=5, pady=5)
delusr = Button(editusers,
text='Del User...',
command=lambda: self.delUser())
delusr.grid(row=0, column=1, padx=5, pady=5)
userpanel = LabelFrame(usersFrame, text="Users", padx=10, pady=10)
userpanel.pack(side=TOP, fill=BOTH)
self.uText = self.getUserText()
self.users_listlabel = Label(userpanel,
text=self.uText,
padx=10,
pady=10)
self.users_listlabel.pack(side=LEFT)
groupspanel = LabelFrame(usersFrame, text="Groups", padx=10, pady=10)
groupspanel.pack(side=TOP, fill=BOTH)
self.gText = self.getGroupText()
self.groups_text = CustomText(groupspanel)
self.groups_text.resetText(self.gText)
self.groups_text.type(DISABLED)
self.groups_text.pack(fill=BOTH)
# Firewall Label
firewall_label = Label(firewallFrame,
text='Firewall Settings',
font=self.liberation_font_15)
firewall_label.pack()
edFrame = Frame(firewallFrame)
fwEnable = Button(edFrame,
text='Enable',
command=lambda: self.enableFirewall())
fwEnable.pack(side=LEFT, padx=10, pady=10, fill=X)
fwDisable = Button(edFrame,
text='Disable',
command=lambda: self.disableFirewall())
fwDisable.pack(side=RIGHT, padx=10, pady=10, fill=X)
edFrame.pack()
firewallpanel = LabelFrame(firewallFrame,
text='Firewall Status',
height=100,
width=450,
padx=10,
pady=10)
firewallpanel.pack(side=TOP, fill=X)
self.fText = self.getFirewallStatus()
self.firewall_text = CustomText(firewallpanel)
self.firewall_text.resetText(self.fText)
self.firewall_text.type(DISABLED)
self.firewall_text.pack(fill=X)
# Services Pane
services_label = Label(servicesFrame,
text='System Services',
font=self.liberation_font_15)
services_label.pack()
servicespanel = LabelFrame(servicesFrame,
text="Services",
padx=10,
pady=10)
servicespanel.pack(side=TOP, fill=BOTH)
self.sText = self.getServicesText()
self.services_text = CustomText(servicespanel)
self.services_text.resetText(self.sText)
self.services_text.type(DISABLED)
self.services_text.pack(fill=BOTH)
# Processes Pane
processes_label = Label(processesFrame,
text='System Processes',
font=self.liberation_font_15)
processes_label.pack()
processespanel = LabelFrame(processesFrame,
text='Processes',
padx=10,
pady=10)
processespanel.pack(side=TOP, fill=BOTH)
self.pText = self.getProcessesText()
self.processes_text = CustomText(processespanel)
self.processes_text.resetText(self.pText)
self.processes_text.type(DISABLED)
self.processes_text.pack(fill=BOTH)
self.root.bind('<Control-r>', self.refresh)
self.root.mainloop()
class InitGUI():
liberation_font_8 = ("Liberation Sans", 8)
liberation_font_10 = ("Liberation Sans", 10)
liberation_font_15 = ("Liberation Sans", 15)
def __init__(self):
self.enc = Encryption()
self.grp = RootPasswordWindow(self.enc)
self.build()
def build(self):
self.root = Tk()
self.root.geometry("800x622+200+200")
self.root.title("SecureMe")
self.root.resizable(width=FALSE, height=FALSE)
menubar = Menu(self.root)
optionsmenu = Menu(menubar, tearoff=0)
optionsmenu.add_command(label="Refresh (Ctrl+r)",
command=lambda: self.refresh("NONE"))
optionsmenu.add_command(label="Exit", command=lambda: self.quitMenu())
menubar.add_cascade(label="Options", menu=optionsmenu)
self.root.config(menu=menubar)
self.toolbar = ToolBar(self.root, self)
style = Style()
style.configure("BW.TLabel",
foreground="black",
background="slate gray",
borderwidth=2)
mFrame = Frame(self.root, bg="slate gray", padx=25, pady=25)
mFrame.pack(fill=BOTH)
self.notebook = ttk.Notebook(mFrame, height=500, style="BW.TLabel")
primaryFrame = Frame(self.notebook, padx=25, pady=25)
usersFrame = Frame(self.notebook, padx=25, pady=25)
firewallFrame = Frame(self.notebook, padx=25, pady=25)
servicesFrame = Frame(self.notebook, padx=25, pady=25)
processesFrame = Frame(self.notebook, padx=25, pady=25)
self.notebook.add(primaryFrame, text='Primary')
self.notebook.add(usersFrame, text='Users')
self.notebook.add(firewallFrame, text='Firewall')
self.notebook.add(servicesFrame, text='Services')
self.notebook.add(processesFrame, text='Processes')
self.notebook.pack(fill=X)
self.updatebar = Frame(self.root)
self.updatebar.pack(side=BOTTOM, fill=X)
self.left_label = Label(self.updatebar, text="Status: None")
self.left_label.pack(side=LEFT, fill=X)
# Primary Panel
primary_label = Label(primaryFrame,
text='Primary Settings',
font=self.liberation_font_15)
primary_label.grid(row=0, column=0, columnspan=2, sticky=N + S + E + W)
actionspanel = LabelFrame(primaryFrame,
text='System Actions',
padx=10,
pady=10)
actionspanel.grid(row=1, column=0, sticky=E + N, padx=25, pady=25)
openterminal = Button(actionspanel,
text='Open Terminal',
command=lambda: self.openTerminal())
openterminal.pack(padx=5, pady=5)
opencontrol = Button(actionspanel,
text='Open Control Panel',
command=lambda: self.openControlPanel())
opencontrol.pack(padx=5, pady=5)
shutdown = Button(actionspanel,
text='Shutdown',
command=lambda: self.shutdown())
shutdown.pack(padx=5, pady=5)
rebootButton = Button(actionspanel,
text='Reboot',
command=lambda: self.reboot())
rebootButton.pack(padx=5, pady=5)
updatespanel = LabelFrame(primaryFrame,
text='System Updates',
padx=10,
pady=10)
updatespanel.grid(row=1, column=1, sticky=W + N, padx=25, pady=25)
update_button = Button(updatespanel,
text='Basic Update',
command=lambda: self.basicUpdate())
update_button.pack(padx=5, pady=5)
upgrade_button = Button(updatespanel,
text='Basic Upgrade',
command=lambda: self.basicUpgrade())
upgrade_button.pack(padx=5, pady=5)
packageupdate_button = Button(updatespanel,
text='Package Update',
command=lambda: self.packageUpdate())
packageupdate_button.pack(padx=5, pady=5)
# Users Panel
users_label = Label(usersFrame,
text='User Security Settings',
font=self.liberation_font_15)
users_label.pack()
editusers = Frame(usersFrame)
editusers.pack()
addusr = Button(editusers,
text='Add User...',
command=lambda: self.addUser())
addusr.grid(row=0, column=0, padx=5, pady=5)
delusr = Button(editusers,
text='Del User...',
command=lambda: self.delUser())
delusr.grid(row=0, column=1, padx=5, pady=5)
userpanel = LabelFrame(usersFrame, text="Users", padx=10, pady=10)
userpanel.pack(side=TOP, fill=BOTH)
self.uText = self.getUserText()
self.users_listlabel = Label(userpanel,
text=self.uText,
padx=10,
pady=10)
self.users_listlabel.pack(side=LEFT)
groupspanel = LabelFrame(usersFrame, text="Groups", padx=10, pady=10)
groupspanel.pack(side=TOP, fill=BOTH)
self.gText = self.getGroupText()
self.groups_text = CustomText(groupspanel)
self.groups_text.resetText(self.gText)
self.groups_text.type(DISABLED)
self.groups_text.pack(fill=BOTH)
# Firewall Label
firewall_label = Label(firewallFrame,
text='Firewall Settings',
font=self.liberation_font_15)
firewall_label.pack()
edFrame = Frame(firewallFrame)
fwEnable = Button(edFrame,
text='Enable',
command=lambda: self.enableFirewall())
fwEnable.pack(side=LEFT, padx=10, pady=10, fill=X)
fwDisable = Button(edFrame,
text='Disable',
command=lambda: self.disableFirewall())
fwDisable.pack(side=RIGHT, padx=10, pady=10, fill=X)
edFrame.pack()
firewallpanel = LabelFrame(firewallFrame,
text='Firewall Status',
height=100,
width=450,
padx=10,
pady=10)
firewallpanel.pack(side=TOP, fill=X)
self.fText = self.getFirewallStatus()
self.firewall_text = CustomText(firewallpanel)
self.firewall_text.resetText(self.fText)
self.firewall_text.type(DISABLED)
self.firewall_text.pack(fill=X)
# Services Pane
services_label = Label(servicesFrame,
text='System Services',
font=self.liberation_font_15)
services_label.pack()
servicespanel = LabelFrame(servicesFrame,
text="Services",
padx=10,
pady=10)
servicespanel.pack(side=TOP, fill=BOTH)
self.sText = self.getServicesText()
self.services_text = CustomText(servicespanel)
self.services_text.resetText(self.sText)
self.services_text.type(DISABLED)
self.services_text.pack(fill=BOTH)
# Processes Pane
processes_label = Label(processesFrame,
text='System Processes',
font=self.liberation_font_15)
processes_label.pack()
processespanel = LabelFrame(processesFrame,
text='Processes',
padx=10,
pady=10)
processespanel.pack(side=TOP, fill=BOTH)
self.pText = self.getProcessesText()
self.processes_text = CustomText(processespanel)
self.processes_text.resetText(self.pText)
self.processes_text.type(DISABLED)
self.processes_text.pack(fill=BOTH)
self.root.bind('<Control-r>', self.refresh)
self.root.mainloop()
def refresh(self, e):
self.setLeftLabel("Refreshing...")
self.uText = self.getUserText()
self.gText = self.getGroupText()
self.sText = self.getServicesText()
self.fText = self.getFirewallStatus()
self.pText = self.getProcessesText()
self.users_listlabel.config(text=self.uText)
self.groups_text.type(NORMAL)
self.groups_text.resetText(self.gText)
self.groups_text.type(DISABLED)
self.services_text.type(NORMAL)
self.services_text.resetText(self.sText)
self.services_text.type(DISABLED)
self.firewall_text.type(NORMAL)
self.firewall_text.resetText(self.fText)
self.firewall_text.type(DISABLED)
self.processes_text.type(NORMAL)
self.processes_text.resetText(self.pText)
self.processes_text.type(DISABLED)
self.resetLeftLabel()
def setLeftLabel(self, s):
self.left_label.config(text=("Status: " + s))
self.root.update()
def resetLeftLabel(self):
self.left_label.config(text="Status: None")
self.root.update()
def getPassword(self):
pwd = self.enc.decrypt()
return pwd
def getUserText(self):
self.setLeftLabel("Getting Users...")
u = Users()
retstr = u.getUsers()
ret = ''
for i in retstr:
ret += "User: "******"\n"
self.resetLeftLabel()
return ret
def getGroupText(self):
self.setLeftLabel("Getting Groups...")
g = Groups()
retstr = g.getGroups()
ret = ''
for i in retstr:
ret += i + "\n"
self.resetLeftLabel()
return ret
def getServicesText(self):
self.setLeftLabel("Getting Services...")
s = Services()
retstr = s.getservicesbasic()
self.resetLeftLabel()
return retstr
def getFirewallStatus(self):
self.setLeftLabel("Getting Firewall Status...")
f = Firewall()
retstr = f.getStatus(self.enc.decrypt())
self.resetLeftLabel()
return retstr
def getProcessesText(self):
self.setLeftLabel("Getting Processes...")
p = Processes()
retstr = p.getprocesses()
self.resetLeftLabel()
return retstr
def basicUpdate(self):
if tkMessageBox.askyesno("SecureMe - Update",
"Proceed with update?") == True:
self.setLeftLabel("Updating Machine...")
ud = Update()
ud.update(self.enc)
self.resetLeftLabel()
else:
pass
def basicUpgrade(self):
if tkMessageBox.askyesno("SecureMe - Upgrade",
"Proceed with upgrade?") == True:
self.setLeftLabel("Upgrading Machine...")
ud = Update()
ud.upgrade(self.enc)
self.resetLeftLabel()
else:
pass
def packageUpdate(self):
if tkMessageBox.askyesno("SecureMe - Package Update",
"Proceed with package update?") == True:
self.setLeftLabel("Updating Packages...")
ud = Update()
ud.updateall(self.enc)
self.resetLeftLabel()
else:
pass
def enableFirewall(self):
if tkMessageBox.askyesno(
"SecureMe - Firewall",
"Are you sure you want to enable the firewall?") == True:
self.setLeftLabel("Enabling Firewall...")
f = Firewall()
f.enable(self.getPassword())
self.refresh("NONE")
self.resetLeftLabel()
else:
pass
def disableFirewall(self):
if tkMessageBox.askyesno(
"SecureMe - Firewall",
"Are you sure you want to disable the firewall?") == True:
self.setLeftLabel("Disabling Firewall...")
f = Firewall()
f.disable(self.getPassword())
self.refresh("NONE")
else:
pass
def openTerminal(self):
self.setLeftLabel("Opening Terminal...")
s = Linux()
s.terminal()
self.resetLeftLabel()
def openControlPanel(self):
self.setLeftLabel("Opening Control Panel")
s = Linux()
s.systemsettings()
self.resetLeftLabel()
def shutdown(self):
if tkMessageBox.askyesno(
"SecureMe - Power",
"Are you sure you would like to power off?") == True:
self.setLeftLabel("Shutting Down...")
s = Linux()
s.shutdown(2)
self.resetLeftLabel()
else:
pass
def reboot(self):
if tkMessageBox.askyesno(
"SecureMe - Power",
"Are you sure you would like to reboot?") == True:
self.setLeftLabel("Rebooting...")
s = Linux()
s.reboot()
self.resetLeftLabel()
else:
pass
def addUser(self):
self.setLeftLabel("Adding user...")
a = AddUser()
def delUser(self):
self.setLeftLabel("Deleting user...")
def quitMenu(self):
if tkMessageBox.askyesno("Secure Me - Quit?",
"Are you sure you want to quit?") == True:
sys.exit(0)
else:
pass
def build(self):
self.root = Tk()
self.root.geometry("800x622+200+200")
self.root.title("SecureMe")
self.root.resizable(width=FALSE, height=FALSE)
menubar = Menu(self.root)
optionsmenu = Menu(menubar, tearoff=0)
optionsmenu.add_command(label="Refresh (Ctrl+r)", command=lambda : self.refresh("NONE"))
optionsmenu.add_command(label="Exit", command=lambda : self.quitMenu())
menubar.add_cascade(label="Options", menu=optionsmenu)
self.root.config(menu=menubar)
self.toolbar = ToolBar(self.root, self)
style = Style()
style.configure("BW.TLabel", foreground="black", background="slate gray", borderwidth=2)
mFrame = Frame(self.root, bg="slate gray", padx=25, pady=25)
mFrame.pack(fill=BOTH)
self.notebook = ttk.Notebook(mFrame, height=500, style="BW.TLabel")
primaryFrame = Frame(self.notebook, padx=25, pady=25)
usersFrame = Frame(self.notebook, padx=25, pady=25)
firewallFrame = Frame(self.notebook, padx=25, pady=25)
servicesFrame = Frame(self.notebook, padx=25, pady=25)
processesFrame = Frame(self.notebook, padx=25, pady=25)
self.notebook.add(primaryFrame, text='Primary')
self.notebook.add(usersFrame, text='Users')
self.notebook.add(firewallFrame, text='Firewall')
self.notebook.add(servicesFrame, text='Services')
self.notebook.add(processesFrame, text='Processes')
self.notebook.pack(fill=X)
self.updatebar = Frame(self.root)
self.updatebar.pack(side=BOTTOM, fill=X)
self.left_label = Label(self.updatebar, text="Status: None")
self.left_label.pack(side=LEFT, fill=X)
# Primary Panel
primary_label = Label(primaryFrame, text='Primary Settings', font=self.liberation_font_15)
primary_label.grid(row=0, column=0, columnspan=2, sticky=N+S+E+W)
actionspanel = LabelFrame(primaryFrame, text='System Actions', padx=10, pady=10)
actionspanel.grid(row=1, column=0, sticky=E+N, padx=25, pady=25)
openterminal = Button(actionspanel, text='Open Terminal', command=lambda : self.openTerminal())
openterminal.pack(padx=5, pady=5)
opencontrol = Button(actionspanel, text='Open Control Panel', command=lambda : self.openControlPanel())
opencontrol.pack(padx=5, pady=5)
shutdown = Button(actionspanel, text='Shutdown', command=lambda : self.shutdown())
shutdown.pack(padx=5, pady=5)
rebootButton = Button(actionspanel, text='Reboot', command=lambda : self.reboot())
rebootButton.pack(padx=5, pady=5)
updatespanel = LabelFrame(primaryFrame, text='System Updates', padx=10, pady=10)
updatespanel.grid(row=1, column=1, sticky=W+N, padx=25, pady=25)
update_button = Button(updatespanel, text='Basic Update', command=lambda : self.basicUpdate())
update_button.pack(padx=5, pady=5)
upgrade_button = Button(updatespanel, text='Basic Upgrade', command=lambda : self.basicUpgrade())
upgrade_button.pack(padx=5, pady=5)
packageupdate_button = Button(updatespanel, text='Package Update', command=lambda : self.packageUpdate())
packageupdate_button.pack(padx=5, pady=5)
# Users Panel
users_label = Label(usersFrame, text='User Security Settings', font=self.liberation_font_15)
users_label.pack()
editusers = Frame(usersFrame)
editusers.pack()
addusr = Button(editusers, text='Add User...', command=lambda : self.addUser())
addusr.grid(row=0, column=0, padx=5, pady=5)
delusr = Button(editusers, text='Del User...', command=lambda : self.delUser())
delusr.grid(row=0, column=1, padx=5, pady=5)
userpanel = LabelFrame(usersFrame, text="Users", padx=10, pady=10)
userpanel.pack(side=TOP, fill=BOTH)
self.uText = self.getUserText()
self.users_listlabel = Label(userpanel, text=self.uText, padx=10, pady=10)
self.users_listlabel.pack(side=LEFT)
groupspanel = LabelFrame(usersFrame, text="Groups", padx=10, pady=10)
groupspanel.pack(side=TOP, fill=BOTH)
self.gText = self.getGroupText()
self.groups_text = CustomText(groupspanel)
self.groups_text.resetText(self.gText)
self.groups_text.type(DISABLED)
self.groups_text.pack(fill=BOTH)
# Firewall Label
firewall_label = Label(firewallFrame, text='Firewall Settings', font=self.liberation_font_15)
firewall_label.pack()
edFrame = Frame(firewallFrame)
fwEnable = Button(edFrame, text='Enable', command=lambda : self.enableFirewall())
fwEnable.pack(side=LEFT, padx=10, pady=10, fill=X)
fwDisable = Button(edFrame, text='Disable', command=lambda : self.disableFirewall())
fwDisable.pack(side=RIGHT, padx=10, pady=10, fill=X)
edFrame.pack()
firewallpanel = LabelFrame(firewallFrame, text='Firewall Status', height=100, width=450, padx=10, pady=10)
firewallpanel.pack(side=TOP, fill=X)
self.fText = self.getFirewallStatus()
self.firewall_text = CustomText(firewallpanel)
self.firewall_text.resetText(self.fText)
self.firewall_text.type(DISABLED)
self.firewall_text.pack(fill=X)
# Services Pane
services_label = Label(servicesFrame, text='System Services', font=self.liberation_font_15)
services_label.pack()
servicespanel = LabelFrame(servicesFrame, text="Services", padx=10, pady=10)
servicespanel.pack(side=TOP, fill=BOTH)
self.sText = self.getServicesText()
self.services_text = CustomText(servicespanel)
self.services_text.resetText(self.sText)
self.services_text.type(DISABLED)
self.services_text.pack(fill=BOTH)
# Processes Pane
processes_label = Label(processesFrame, text='System Processes', font=self.liberation_font_15)
processes_label.pack()
processespanel = LabelFrame(processesFrame, text='Processes', padx=10, pady=10)
processespanel.pack(side=TOP, fill=BOTH)
self.pText = self.getProcessesText()
self.processes_text = CustomText(processespanel)
self.processes_text.resetText(self.pText)
self.processes_text.type(DISABLED)
self.processes_text.pack(fill=BOTH)
self.root.bind('<Control-r>', self.refresh)
self.root.mainloop()
class InitGUI():
liberation_font_8 = ("Liberation Sans", 8)
liberation_font_10 = ("Liberation Sans", 10)
liberation_font_15 = ("Liberation Sans", 15)
def __init__(self):
self.enc = Encryption()
self.grp = RootPasswordWindow(self.enc)
self.build()
def build(self):
self.root = Tk()
self.root.geometry("800x622+200+200")
self.root.title("SecureMe")
self.root.resizable(width=FALSE, height=FALSE)
menubar = Menu(self.root)
optionsmenu = Menu(menubar, tearoff=0)
optionsmenu.add_command(label="Refresh (Ctrl+r)", command=lambda : self.refresh("NONE"))
optionsmenu.add_command(label="Exit", command=lambda : self.quitMenu())
menubar.add_cascade(label="Options", menu=optionsmenu)
self.root.config(menu=menubar)
self.toolbar = ToolBar(self.root, self)
style = Style()
style.configure("BW.TLabel", foreground="black", background="slate gray", borderwidth=2)
mFrame = Frame(self.root, bg="slate gray", padx=25, pady=25)
mFrame.pack(fill=BOTH)
self.notebook = ttk.Notebook(mFrame, height=500, style="BW.TLabel")
primaryFrame = Frame(self.notebook, padx=25, pady=25)
usersFrame = Frame(self.notebook, padx=25, pady=25)
firewallFrame = Frame(self.notebook, padx=25, pady=25)
servicesFrame = Frame(self.notebook, padx=25, pady=25)
processesFrame = Frame(self.notebook, padx=25, pady=25)
self.notebook.add(primaryFrame, text='Primary')
self.notebook.add(usersFrame, text='Users')
self.notebook.add(firewallFrame, text='Firewall')
self.notebook.add(servicesFrame, text='Services')
self.notebook.add(processesFrame, text='Processes')
self.notebook.pack(fill=X)
self.updatebar = Frame(self.root)
self.updatebar.pack(side=BOTTOM, fill=X)
self.left_label = Label(self.updatebar, text="Status: None")
self.left_label.pack(side=LEFT, fill=X)
# Primary Panel
primary_label = Label(primaryFrame, text='Primary Settings', font=self.liberation_font_15)
primary_label.grid(row=0, column=0, columnspan=2, sticky=N+S+E+W)
actionspanel = LabelFrame(primaryFrame, text='System Actions', padx=10, pady=10)
actionspanel.grid(row=1, column=0, sticky=E+N, padx=25, pady=25)
openterminal = Button(actionspanel, text='Open Terminal', command=lambda : self.openTerminal())
openterminal.pack(padx=5, pady=5)
opencontrol = Button(actionspanel, text='Open Control Panel', command=lambda : self.openControlPanel())
opencontrol.pack(padx=5, pady=5)
shutdown = Button(actionspanel, text='Shutdown', command=lambda : self.shutdown())
shutdown.pack(padx=5, pady=5)
rebootButton = Button(actionspanel, text='Reboot', command=lambda : self.reboot())
rebootButton.pack(padx=5, pady=5)
updatespanel = LabelFrame(primaryFrame, text='System Updates', padx=10, pady=10)
updatespanel.grid(row=1, column=1, sticky=W+N, padx=25, pady=25)
update_button = Button(updatespanel, text='Basic Update', command=lambda : self.basicUpdate())
update_button.pack(padx=5, pady=5)
upgrade_button = Button(updatespanel, text='Basic Upgrade', command=lambda : self.basicUpgrade())
upgrade_button.pack(padx=5, pady=5)
packageupdate_button = Button(updatespanel, text='Package Update', command=lambda : self.packageUpdate())
packageupdate_button.pack(padx=5, pady=5)
# Users Panel
users_label = Label(usersFrame, text='User Security Settings', font=self.liberation_font_15)
users_label.pack()
editusers = Frame(usersFrame)
editusers.pack()
addusr = Button(editusers, text='Add User...', command=lambda : self.addUser())
addusr.grid(row=0, column=0, padx=5, pady=5)
delusr = Button(editusers, text='Del User...', command=lambda : self.delUser())
delusr.grid(row=0, column=1, padx=5, pady=5)
userpanel = LabelFrame(usersFrame, text="Users", padx=10, pady=10)
userpanel.pack(side=TOP, fill=BOTH)
self.uText = self.getUserText()
self.users_listlabel = Label(userpanel, text=self.uText, padx=10, pady=10)
self.users_listlabel.pack(side=LEFT)
groupspanel = LabelFrame(usersFrame, text="Groups", padx=10, pady=10)
groupspanel.pack(side=TOP, fill=BOTH)
self.gText = self.getGroupText()
self.groups_text = CustomText(groupspanel)
self.groups_text.resetText(self.gText)
self.groups_text.type(DISABLED)
self.groups_text.pack(fill=BOTH)
# Firewall Label
firewall_label = Label(firewallFrame, text='Firewall Settings', font=self.liberation_font_15)
firewall_label.pack()
edFrame = Frame(firewallFrame)
fwEnable = Button(edFrame, text='Enable', command=lambda : self.enableFirewall())
fwEnable.pack(side=LEFT, padx=10, pady=10, fill=X)
fwDisable = Button(edFrame, text='Disable', command=lambda : self.disableFirewall())
fwDisable.pack(side=RIGHT, padx=10, pady=10, fill=X)
edFrame.pack()
firewallpanel = LabelFrame(firewallFrame, text='Firewall Status', height=100, width=450, padx=10, pady=10)
firewallpanel.pack(side=TOP, fill=X)
self.fText = self.getFirewallStatus()
self.firewall_text = CustomText(firewallpanel)
self.firewall_text.resetText(self.fText)
self.firewall_text.type(DISABLED)
self.firewall_text.pack(fill=X)
# Services Pane
services_label = Label(servicesFrame, text='System Services', font=self.liberation_font_15)
services_label.pack()
servicespanel = LabelFrame(servicesFrame, text="Services", padx=10, pady=10)
servicespanel.pack(side=TOP, fill=BOTH)
self.sText = self.getServicesText()
self.services_text = CustomText(servicespanel)
self.services_text.resetText(self.sText)
self.services_text.type(DISABLED)
self.services_text.pack(fill=BOTH)
# Processes Pane
processes_label = Label(processesFrame, text='System Processes', font=self.liberation_font_15)
processes_label.pack()
processespanel = LabelFrame(processesFrame, text='Processes', padx=10, pady=10)
processespanel.pack(side=TOP, fill=BOTH)
self.pText = self.getProcessesText()
self.processes_text = CustomText(processespanel)
self.processes_text.resetText(self.pText)
self.processes_text.type(DISABLED)
self.processes_text.pack(fill=BOTH)
self.root.bind('<Control-r>', self.refresh)
self.root.mainloop()
def refresh(self, e):
self.setLeftLabel("Refreshing...")
self.uText = self.getUserText()
self.gText = self.getGroupText()
self.sText = self.getServicesText()
self.fText = self.getFirewallStatus()
self.pText = self.getProcessesText()
self.users_listlabel.config(text=self.uText)
self.groups_text.type(NORMAL)
self.groups_text.resetText(self.gText)
self.groups_text.type(DISABLED)
self.services_text.type(NORMAL)
self.services_text.resetText(self.sText)
self.services_text.type(DISABLED)
self.firewall_text.type(NORMAL)
self.firewall_text.resetText(self.fText)
self.firewall_text.type(DISABLED)
self.processes_text.type(NORMAL)
self.processes_text.resetText(self.pText)
self.processes_text.type(DISABLED)
self.resetLeftLabel()
def setLeftLabel(self, s):
self.left_label.config(text=("Status: "+s))
self.root.update()
def resetLeftLabel(self):
self.left_label.config(text="Status: None")
self.root.update()
def getPassword(self):
pwd = self.enc.decrypt()
return pwd
def getUserText(self):
self.setLeftLabel("Getting Users...")
u = Users()
retstr = u.getUsers()
ret = ''
for i in retstr:
ret += "User: "******"\n"
self.resetLeftLabel()
return ret
def getGroupText(self):
self.setLeftLabel("Getting Groups...")
g = Groups()
retstr = g.getGroups()
ret = ''
for i in retstr:
ret += i + "\n"
self.resetLeftLabel()
return ret
def getServicesText(self):
self.setLeftLabel("Getting Services...")
s = Services()
retstr = s.getservicesbasic()
self.resetLeftLabel()
return retstr
def getFirewallStatus(self):
self.setLeftLabel("Getting Firewall Status...")
f = Firewall()
retstr = f.getStatus(self.enc.decrypt())
self.resetLeftLabel()
return retstr
def getProcessesText(self):
self.setLeftLabel("Getting Processes...")
p = Processes()
retstr = p.getprocesses()
self.resetLeftLabel()
return retstr
def basicUpdate(self):
if tkMessageBox.askyesno("SecureMe - Update", "Proceed with update?") == True:
self.setLeftLabel("Updating Machine...")
ud = Update()
ud.update(self.enc)
self.resetLeftLabel()
else:
pass
def basicUpgrade(self):
if tkMessageBox.askyesno("SecureMe - Upgrade", "Proceed with upgrade?") == True:
self.setLeftLabel("Upgrading Machine...")
ud = Update()
ud.upgrade(self.enc)
self.resetLeftLabel()
else:
pass
def packageUpdate(self):
if tkMessageBox.askyesno("SecureMe - Package Update", "Proceed with package update?") == True:
self.setLeftLabel("Updating Packages...")
ud = Update()
ud.updateall(self.enc)
self.resetLeftLabel()
else:
pass
def enableFirewall(self):
if tkMessageBox.askyesno("SecureMe - Firewall", "Are you sure you want to enable the firewall?") == True:
self.setLeftLabel("Enabling Firewall...")
f = Firewall()
f.enable(self.getPassword())
self.refresh("NONE")
self.resetLeftLabel()
else:
pass
def disableFirewall(self):
if tkMessageBox.askyesno("SecureMe - Firewall", "Are you sure you want to disable the firewall?") == True:
self.setLeftLabel("Disabling Firewall...")
f = Firewall()
f.disable(self.getPassword())
self.refresh("NONE")
else:
pass
def openTerminal(self):
self.setLeftLabel("Opening Terminal...")
s = Linux()
s.terminal()
self.resetLeftLabel()
def openControlPanel(self):
self.setLeftLabel("Opening Control Panel")
s = Linux()
s.systemsettings()
self.resetLeftLabel()
def shutdown(self):
if tkMessageBox.askyesno("SecureMe - Power", "Are you sure you would like to power off?") == True:
self.setLeftLabel("Shutting Down...")
s = Linux()
s.shutdown(2)
self.resetLeftLabel()
else:
pass
def reboot(self):
if tkMessageBox.askyesno("SecureMe - Power", "Are you sure you would like to reboot?") == True:
self.setLeftLabel("Rebooting...")
s = Linux()
s.reboot()
self.resetLeftLabel()
else:
pass
def addUser(self):
self.setLeftLabel("Adding user...")
a = AddUser()
def delUser(self):
self.setLeftLabel("Deleting user...")
def quitMenu(self):
if tkMessageBox.askyesno("Secure Me - Quit?", "Are you sure you want to quit?") == True:
sys.exit(0)
else:
pass
class Dataset(Dataset):
def __init__(self,
filename,
preprocess_config,
train_config,
model_config,
sort=False,
drop_last=False):
self.dataset_name = preprocess_config["dataset"]
self.preprocessed_path = preprocess_config["path"]["preprocessed_path"]
self.cleaners = preprocess_config["preprocessing"]["text"][
"text_cleaners"]
self.batch_size = train_config["optimizer"]["batch_size"]
self.ct = CustomText(model_config['level'])
# getattr(self, 'text_to_sequence', ct.text_to_sequence)
self.basename, self.speaker, self.text, self.raw_text = self.process_meta(
filename)
with open(os.path.join(self.preprocessed_path, "speakers.json")) as f:
self.speaker_map = json.load(f)
self.sort = sort
self.drop_last = drop_last
def __len__(self):
return len(self.text)
def __getitem__(self, idx):
basename = self.basename[idx]
speaker = self.speaker[idx]
speaker_id = self.speaker_map[speaker]
raw_text = self.raw_text[idx]
phone = np.array(
self.ct.text_to_sequence(self.text[idx], self.cleaners))
mel_path = os.path.join(
self.preprocessed_path,
"mel",
"{}-mel-{}.npy".format(speaker, basename),
)
mel = np.load(mel_path)
pitch_path = os.path.join(
self.preprocessed_path,
"pitch",
"{}-pitch-{}.npy".format(speaker, basename),
)
pitch = np.load(pitch_path)
energy_path = os.path.join(
self.preprocessed_path,
"energy",
"{}-energy-{}.npy".format(speaker, basename),
)
energy = np.load(energy_path)
duration_path = os.path.join(
self.preprocessed_path,
"duration",
"{}-duration-{}.npy".format(speaker, basename),
)
duration = np.load(duration_path)
sample = {
"id": basename,
"speaker": speaker_id,
"text": phone,
"raw_text": raw_text,
"mel": mel,
"pitch": pitch,
"energy": energy,
"duration": duration,
}
return sample
def process_meta(self, filename):
with open(os.path.join(self.preprocessed_path, filename),
"r",
encoding="utf-8") as f:
name = []
speaker = []
text = []
raw_text = []
for line in f.readlines():
n, s, t, r = line.strip("\n").split("|")
name.append(n)
speaker.append(s)
text.append(t)
raw_text.append(r)
return name, speaker, text, raw_text
def reprocess(self, data, idxs):
ids = [data[idx]["id"] for idx in idxs]
speakers = [data[idx]["speaker"] for idx in idxs]
texts = [data[idx]["text"] for idx in idxs]
raw_texts = [data[idx]["raw_text"] for idx in idxs]
mels = [data[idx]["mel"] for idx in idxs]
pitches = [data[idx]["pitch"] for idx in idxs]
energies = [data[idx]["energy"] for idx in idxs]
durations = [data[idx]["duration"] for idx in idxs]
text_lens = np.array([text.shape[0] for text in texts])
mel_lens = np.array([mel.shape[0] for mel in mels])
speakers = np.array(speakers)
texts = pad_1D(texts)
mels = pad_2D(mels)
pitches = pad_1D(pitches)
energies = pad_1D(energies)
durations = pad_1D(durations)
return (
ids,
raw_texts,
speakers,
texts,
text_lens,
max(text_lens),
mels,
mel_lens,
max(mel_lens),
pitches,
energies,
durations,
)
def collate_fn(self, data):
data_size = len(data)
if self.sort:
len_arr = np.array([d["text"].shape[0] for d in data])
idx_arr = np.argsort(-len_arr)
else:
idx_arr = np.arange(data_size)
tail = idx_arr[len(idx_arr) - (len(idx_arr) % self.batch_size):]
idx_arr = idx_arr[:len(idx_arr) - (len(idx_arr) % self.batch_size)]
idx_arr = idx_arr.reshape((-1, self.batch_size)).tolist()
if not self.drop_last and len(tail) > 0:
idx_arr += [tail.tolist()]
output = list()
for idx in idx_arr:
output.append(self.reprocess(data, idx))
return output
