62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
|
import subprocess
|
||
|
from PyInquirer import prompt
|
||
|
from AbstractModule import AbstractModule
|
||
|
|
||
|
|
||
|
class VimModule(AbstractModule):
|
||
|
def __init__(self):
|
||
|
super().__init__()
|
||
|
|
||
|
|
||
|
# Check default swap location
|
||
|
def convert_string(string):
|
||
|
li = list(string.split())
|
||
|
return li
|
||
|
|
||
|
|
||
|
output_swaps = subprocess.check_output(['cat', '/proc/swaps']).decode("UTF-8")
|
||
|
final_list = convert_string(output_swaps)
|
||
|
print(final_list[5])
|
||
|
# End of location check
|
||
|
|
||
|
output_swapon = subprocess.check_output(['swapon', '--show']).decode("UTF-8")
|
||
|
output_free = subprocess.check_output(['free', '-h']).decode("UTF-8")
|
||
|
|
||
|
option = input("Enter option. 1.Create/Resize 2.Delete: ")
|
||
|
|
||
|
if option == "1":
|
||
|
if not output_swapon:
|
||
|
size = input("How big should the swap be(numbers in Gigabyte)? ")
|
||
|
create_swapfile = """sudo fallocate -l {}G /swapfile &&
|
||
|
sudo chmod 600 /swapfile &&
|
||
|
sudo mkswap /swapfile &&
|
||
|
sudo swapon /swapfile""".format(size)
|
||
|
subprocess.call(create_swapfile, shell=True)
|
||
|
|
||
|
output_free = subprocess.check_output(['free', '-h']).decode("UTF-8")
|
||
|
print(output_free.strip())
|
||
|
|
||
|
else:
|
||
|
print("")
|
||
|
print("Swap already installed! You can resize it!")
|
||
|
resize = input("How big should your swap become(numbers in Gigabyte)? ")
|
||
|
resize_swapfile = """sudo swapoff /swapfile &&
|
||
|
sudo fallocate -l {}G /swapfile &&
|
||
|
sudo mkswap /swapfile &&
|
||
|
sudo swapon /swapfile""".format(resize)
|
||
|
subprocess.call(resize_swapfile, shell=True)
|
||
|
output_free = subprocess.check_output(['free', '-h']).decode("UTF-8")
|
||
|
print(output_free.strip())
|
||
|
elif option == "2":
|
||
|
disable_swapfile = """sudo swapoff /swapfile &&
|
||
|
sudo rm /swapfile"""
|
||
|
|
||
|
subprocess.call(disable_swapfile, shell=True)
|
||
|
output_swapon = subprocess.check_output(['swapon', '--show']).decode("UTF-8")
|
||
|
output_free = subprocess.check_output(['free', '-h']).decode("UTF-8")
|
||
|
if not output_swapon:
|
||
|
print("Swap deleted!")
|
||
|
print(output_free.strip())
|
||
|
else:
|
||
|
print("Wrong input!")
|