mirror of
https://github.com/bytedream/aion.git
synced 2025-05-10 20:35:07 +02:00
236 lines
8.1 KiB
Python
236 lines
8.1 KiB
Python
#!/usr/bin/python3
|
|
|
|
from glob import glob
|
|
from sys import path
|
|
|
|
path.insert(0, glob("/usr/local/aion-*/aion_core")[0])
|
|
|
|
from aion_core import config as aconf
|
|
from aion_core import language as alang
|
|
|
|
from aion_core.skill import Skill
|
|
|
|
|
|
class CpuRamUsage(Skill):
|
|
"""
|
|
skill class to get the cpu and ram usage
|
|
"""
|
|
|
|
def main(self):
|
|
from psutil import cpu_percent, virtual_memory
|
|
from statistics import mean
|
|
from time import sleep
|
|
|
|
testing_time = 10
|
|
print_process = False
|
|
|
|
cpu_list = []
|
|
ram_list = []
|
|
|
|
while True:
|
|
cpu = cpu_percent()
|
|
cpu_list.append(cpu)
|
|
ram = virtual_memory()[2]
|
|
ram_list.append(ram)
|
|
if print_process is True:
|
|
print("CPU: " + str(cpu) + "%")
|
|
print("RAM: " + str(ram) + "%")
|
|
sleep(1)
|
|
|
|
if 0.0 in cpu_list:
|
|
cpu_list.remove(0.0)
|
|
if len(cpu_list) == testing_time and len(ram_list) == testing_time:
|
|
cpu = mean(cpu_list)
|
|
ram = mean(ram_list)
|
|
break
|
|
elif len(cpu_list) == testing_time - 1 and len(ram_list) == testing_time:
|
|
cpu = cpu_percent()
|
|
cpu_list.append(cpu)
|
|
cpu = mean(cpu_list)
|
|
ram = mean(ram_list)
|
|
break
|
|
|
|
self.speech_output(alang.start("skills", "cpu_ram_usage", {"cpu_usage": round(cpu, 2), "ram_usage": round(ram, 2)}))
|
|
|
|
|
|
class CpuUsage(Skill):
|
|
"""
|
|
skill class to get the cpu usage
|
|
"""
|
|
|
|
def main(self):
|
|
from statistics import mean
|
|
from psutil import cpu_percent
|
|
from time import sleep
|
|
|
|
testing_time = 10
|
|
print_process = False
|
|
|
|
cpu_list = []
|
|
|
|
while True:
|
|
cpu = cpu_percent()
|
|
cpu_list.append(cpu)
|
|
if print_process is True:
|
|
print("CPU: " + str(cpu) + "%")
|
|
sleep(1)
|
|
|
|
if 0.0 in cpu_list:
|
|
cpu_list.remove(0.0)
|
|
if len(cpu_list) == testing_time:
|
|
cpu = (mean(cpu_list))
|
|
break
|
|
|
|
self.speech_output(alang.start("skills", "cpu_usage", {"cpu_usage": round(cpu, 2)}))
|
|
|
|
|
|
class CurrentTime(Skill):
|
|
"""
|
|
skill class to get the current time
|
|
"""
|
|
|
|
def main(self):
|
|
from datetime import datetime
|
|
|
|
current_time = datetime.now()
|
|
time_dict = {"hour": current_time.hour, "minute": current_time.minute, "second": current_time.second, "microsecond": current_time.microsecond}
|
|
if aconf.Aion().get_time_format() == "12":
|
|
if time_dict["minute"] == 0:
|
|
if time_dict["hour"] < 13:
|
|
self.speech_output(alang.start("skills", "current_time_full_am", time_dict))
|
|
else:
|
|
time_dict["hour"] = time_dict["hour"] - 12
|
|
self.speech_output(alang.start("skills", "current_time_full_pm", time_dict))
|
|
elif time_dict["minute"] == 30:
|
|
if time_dict["hour"] < 13:
|
|
self.speech_output(alang.start("skills", "current_time_half_am", time_dict))
|
|
else:
|
|
time_dict["hour"] = time_dict["hour"] - 12
|
|
self.speech_output(alang.start("skills", "current_time_half_pm", time_dict))
|
|
elif 0 < time_dict["minute"] < 30:
|
|
if time_dict["hour"] < 13:
|
|
self.speech_output(alang.start("skills", "current_time_past_am", time_dict))
|
|
else:
|
|
time_dict["hour"] = time_dict["hour"] - 12
|
|
self.speech_output(alang.start("skills", "current_time_past_pm", time_dict))
|
|
elif time_dict["minute"] > 30:
|
|
time_dict["minute"] = 60 - time_dict["minute"]
|
|
if time_dict["hour"] < 13:
|
|
self.speech_output(alang.start("skills", "current_time_to_am", time_dict))
|
|
else:
|
|
time_dict["hour"] = time_dict["hour"] - 12
|
|
self.speech_output(alang.start("skills", "current_time_to_pm", time_dict))
|
|
else:
|
|
self.speech_output(alang.start("skills", "current_time_etc", time_dict))
|
|
else:
|
|
if time_dict["minute"] == 0:
|
|
self.speech_output(alang.start("skills", "current_time_full", time_dict))
|
|
elif time_dict["minute"] == 30:
|
|
self.speech_output(alang.start("skills", "current_time_half", time_dict))
|
|
else:
|
|
self.speech_output(alang.start("skills", "current_time_etc", time_dict))
|
|
|
|
|
|
class IpAddress(Skill):
|
|
"""
|
|
skill class to get the ip address
|
|
"""
|
|
|
|
def main(self):
|
|
from socket import gethostname, gethostbyname
|
|
|
|
self.speech_output(alang.start("skills", "ip_address", {"ip_address": gethostbyname(gethostname() + ".local")}))
|
|
|
|
|
|
class Play(Skill):
|
|
"""
|
|
skill class to play a audio from youtube
|
|
"""
|
|
|
|
def main(self):
|
|
from aion_core.utils import get_full_directory_data, vlc
|
|
|
|
file_or_search_element = self.speech_input.lower().replace(self.activate_phrase.lower(), "")
|
|
for file in get_full_directory_data(file_or_search_element):
|
|
if file_or_search_element in file:
|
|
return vlc(file_or_search_element)
|
|
import urllib.parse, urllib.request
|
|
from pafy import new
|
|
from re import findall
|
|
search_query = urllib.parse.urlencode({"search_query": file_or_search_element})
|
|
for i in range(10): # sometimes the video url's cannot be found
|
|
try:
|
|
html_content = urllib.request.urlopen("https://www.youtube.com/results?" + search_query)
|
|
search_results = findall(r'href=\"\/watch\?v=(.{11})', html_content.read().decode())
|
|
vlc(new(str("https://www.youtube.com/watch?v=" + search_results[0])).getbestaudio().url)
|
|
break
|
|
# i do it this way, because vlc sometimes don't extract the 'only audio url' from the youtube url
|
|
except IndexError:
|
|
pass
|
|
|
|
def run_after(self):
|
|
for plugin in self.run_after_plugins:
|
|
self.start_run_after_plugin(plugin["name"])
|
|
|
|
|
|
class RamUsage(Skill):
|
|
"""
|
|
skill class to get ram usage
|
|
"""
|
|
|
|
def main(self):
|
|
from statistics import mean
|
|
from psutil import virtual_memory
|
|
from time import sleep
|
|
|
|
testing_time = 10
|
|
print_process = False
|
|
|
|
ram_list = []
|
|
|
|
while True:
|
|
ram = virtual_memory()[2]
|
|
ram_list.append(ram)
|
|
if print_process is True:
|
|
print("RAM: " + str(ram) + "%")
|
|
sleep(1)
|
|
|
|
if len(ram_list) == testing_time:
|
|
ram = mean(ram_list)
|
|
break
|
|
|
|
self.speech_output(alang.start("skills", "ram_usage", {"ram_usage": round(ram, 2)}))
|
|
|
|
|
|
class Shutdown(Skill):
|
|
"""
|
|
skill class to shutdown the system
|
|
"""
|
|
|
|
def main(self):
|
|
from os import system
|
|
|
|
system("sudo shutdown -h 0")
|
|
|
|
|
|
class Wikipedia(Skill):
|
|
"""
|
|
skill class to search an article on wikipedia
|
|
"""
|
|
|
|
def main(self):
|
|
from aion_core.utils import remove_brackets, remove_space, remove_string_sequence
|
|
from wikipediaapi import Wikipedia
|
|
|
|
splitted_acph = self.activate_phrase.split("__and__")
|
|
searched_article = remove_string_sequence(self.speech_input, splitted_acph[0], splitted_acph[-1])
|
|
wiki = Wikipedia(aconf.Aion().get_language().split("_")[0])
|
|
article = wiki.page(searched_article)
|
|
|
|
if article.exists():
|
|
article_text = remove_brackets(article.summary)
|
|
article_text = remove_space(article_text)
|
|
self.speech_output(alang.start("skills", "wikipedia", {"article_text": article_text}))
|
|
else:
|
|
self.speech_output(alang.start("skills", "wikipedia_article_not_found", {"article_name": searched_article}))
|