alert-battery-to-maintain-health/src/__main__.py

__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)
NOTIFICATION_PATH_ICON = os.path.abspath(os.path.join(__file__, "..", ".." ,"icons/battery-status.png"))
NOTIFICATION = Notify(
    default_notification_title="Alert battery",
    default_application_name="Alert battery",
    default_notification_icon=NOTIFICATION_PATH_ICON,
)


def get_sensor_battery():
    """Get object data sensor battery"""
    return psutil.sensors_battery()


def send_notification(message, title="Battery"):
    """Send native notification"""
    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__':

    # Is battery
    if get_sensor_battery() :

        # 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)