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'