From 23771e22beddbf2ebdf663b3f278475cd2ce21ea Mon Sep 17 00:00:00 2001
From: Lucas Noki <lucas-noki@gmx.de>
Date: Fri, 17 Jan 2020 19:21:48 +0100
Subject: [PATCH] Initial swap file

---
 modules/swap/module.py | 61 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 modules/swap/module.py

diff --git a/modules/swap/module.py b/modules/swap/module.py
new file mode 100644
index 0000000..519dfe5
--- /dev/null
+++ b/modules/swap/module.py
@@ -0,0 +1,61 @@
+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!")