Sideband_CE/sbapp/plyer/platforms/android/vibrator.py

"""Implementation Vibrator for Android."""

from jnius import autoclass, cast
from plyer.facades import Vibrator
from plyer.platforms.android import activity
from plyer.platforms.android import SDK_INT

Context = autoclass("android.content.Context")
vibrator_service = activity.getSystemService(Context.VIBRATOR_SERVICE)
vibrator = cast("android.os.Vibrator", vibrator_service)
if SDK_INT >= 26:
    VibrationEffect = autoclass("android.os.VibrationEffect")


class AndroidVibrator(Vibrator):
    """Android Vibrator class.

    Supported features:
        * vibrate for some period of time.
        * vibrate from given pattern.
        * cancel vibration.
        * check whether Vibrator exists.
    """

    def _vibrate(self, time=None, **kwargs):
        if vibrator:
            if SDK_INT >= 26:
                vibrator.vibrate(
                    VibrationEffect.createOneShot(
                        int(1000 * time), VibrationEffect.DEFAULT_AMPLITUDE
                    )
                )
            else:
                vibrator.vibrate(int(1000 * time))

    def _pattern(self, pattern=None, repeat=None, **kwargs):
        pattern = [int(1000 * time) for time in pattern]

        if vibrator:
            if SDK_INT >= 26:
                vibrator.vibrate(
                    VibrationEffect.createWaveform(pattern, repeat)
                )
            else:
                vibrator.vibrate(pattern, repeat)

    def _exists(self, **kwargs):
        if SDK_INT >= 11:
            return vibrator.hasVibrator()
        elif vibrator_service is None:
            raise NotImplementedError()
        return True

    def _cancel(self, **kwargs):
        vibrator.cancel()


def instance():
    """Returns Vibrator with android features.

    :return: instance of class AndroidVibrator
    """
    return AndroidVibrator()