From 430cc253a2235f9ca4fa94c9110eb5cc85a366cf Mon Sep 17 00:00:00 2001
From: Andros Fenollosa <andros@fenollosa.email>
Date: Sun, 29 Aug 2021 11:58:28 +0200
Subject: [PATCH] Add code

---
 alert_battery_to_maintain_health/__init__.py  | 71 +++++++++++++++----
 tests/__init__.py                             |  0
 .../test_alert_battery_to_maintain_health.py  |  5 --
 3 files changed, 58 insertions(+), 18 deletions(-)
 delete mode 100755 tests/__init__.py
 delete mode 100755 tests/test_alert_battery_to_maintain_health.py

diff --git a/alert_battery_to_maintain_health/__init__.py b/alert_battery_to_maintain_health/__init__.py
index 2e709a0..192f3f8 100755
--- a/alert_battery_to_maintain_health/__init__.py
+++ b/alert_battery_to_maintain_health/__init__.py
@@ -1,21 +1,66 @@
-__version__ = '0.1.0'
+__version__ = '1.0.0'
 
+# Dependencies
+import os
+from tempfile import gettempdir
 from notifypy import Notify
 import psutil
 
+# Variables
+LIMIT_ABOVE_BATTERY = 85
+MESSAGE_LIMIT_ABOVE_BATTERY = "Disconnect your charger"
+LIMIT_BELOW_BATTERY = 15
+MESSAGE_LIMIT_BELOW_BATTERY = "Connect your charger"
+FILE_NAME_TEMP = "alert_battery_to_maintain_health"
+PATH_FILE_NAME_TEMP = os.path.join(gettempdir(), FILE_NAME_TEMP)
+
+
+def get_sensor_battery():
+    """Get object data sensor battery"""
+    return psutil.sensors_battery()
+
+
+def send_notification(message, title="Battery"):
+    """Send native notification"""
+    notification = Notify()
+    notification.title = title
+    notification.message = message
+    notification.send()
+
+
+def create_file_block():
+    """Make temp file not repeat notification"""
+    open(PATH_FILE_NAME_TEMP, 'w').close()
+
+
+def get_battery_percent():
+    """Gets the percentage of battery charge"""
+    return int(get_sensor_battery().percent)
+
+
+def is_plugged():
+    """It tells you if it is charging or not."""
+    return get_sensor_battery().power_plugged
+
+
 if __name__ == '__main__':
-    # Variables
-    battery = psutil.sensors_battery()
 
-    if battery:
-        notification = Notify()
-        notification.title = "Cool Title"
-        notification.message = "Even cooler message."
-        notification.send()
+    # Is battery
+    if get_sensor_battery() :
 
-        print(battery)
-        plugged = battery.power_plugged
-        percent = str(battery.percent)
-        plugged = "Plugged In" if plugged else "Not Plugged In"
-        print(percent+'% | '+plugged)
+        # It has not been previously warned
+        if not os.path.exists(PATH_FILE_NAME_TEMP):
 
+            # Below
+            if LIMIT_BELOW_BATTERY > get_battery_percent() and not is_plugged():
+                send_notification(MESSAGE_LIMIT_BELOW_BATTERY)
+                create_file_block()
+
+            # Above
+            if LIMIT_ABOVE_BATTERY < get_battery_percent() and is_plugged():
+                send_notification(MESSAGE_LIMIT_ABOVE_BATTERY)
+                create_file_block()
+
+        # Unlock to warned
+        elif LIMIT_BELOW_BATTERY < get_battery_percent() < LIMIT_ABOVE_BATTERY:
+            os.remove(PATH_FILE_NAME_TEMP)
\ No newline at end of file
diff --git a/tests/__init__.py b/tests/__init__.py
deleted file mode 100755
index e69de29..0000000
diff --git a/tests/test_alert_battery_to_maintain_health.py b/tests/test_alert_battery_to_maintain_health.py
deleted file mode 100755
index eef7e7c..0000000
--- a/tests/test_alert_battery_to_maintain_health.py
+++ /dev/null
@@ -1,5 +0,0 @@
-from alert_battery_to_maintain_health import __version__
-
-
-def test_version():
-    assert __version__ == '0.1.0'