|
| 1 | +from abc import ABC, abstractmethod |
| 2 | +from .utils import IS_ANDROID |
| 3 | +from .view import ViewBase |
| 4 | + |
| 5 | +# ======================================== |
| 6 | +# Base class |
| 7 | +# ======================================== |
| 8 | + |
| 9 | + |
| 10 | +class MaterialDatePickerBase(ABC): |
| 11 | + @abstractmethod |
| 12 | + def __init__(self) -> None: |
| 13 | + super().__init__() |
| 14 | + |
| 15 | + @abstractmethod |
| 16 | + def set_date(self, year: int, month: int, day: int) -> None: |
| 17 | + pass |
| 18 | + |
| 19 | + @abstractmethod |
| 20 | + def get_date(self) -> tuple: |
| 21 | + pass |
| 22 | + |
| 23 | + |
| 24 | +if IS_ANDROID: |
| 25 | + # ======================================== |
| 26 | + # Android class |
| 27 | + # ======================================== |
| 28 | + |
| 29 | + from java import jclass |
| 30 | + |
| 31 | + class MaterialDatePicker(MaterialDatePickerBase, ViewBase): |
| 32 | + def __init__(self, year: int = 0, month: int = 0, day: int = 0) -> None: |
| 33 | + super().__init__() |
| 34 | + self.native_class = jclass("com.google.android.material.datepicker.MaterialDatePicker") |
| 35 | + self.builder = self.native_class.Builder.datePicker() |
| 36 | + self.set_date(year, month, day) |
| 37 | + self.native_instance = self.builder.build() |
| 38 | + |
| 39 | + def set_date(self, year: int, month: int, day: int) -> None: |
| 40 | + # MaterialDatePicker uses milliseconds since epoch to set date |
| 41 | + from java.util import Calendar |
| 42 | + cal = Calendar.getInstance() |
| 43 | + cal.set(year, month, day) |
| 44 | + milliseconds = cal.getTimeInMillis() |
| 45 | + self.builder.setSelection(milliseconds) |
| 46 | + |
| 47 | + def get_date(self) -> tuple: |
| 48 | + # Convert selection (milliseconds since epoch) back to a date |
| 49 | + from java.util import Calendar |
| 50 | + cal = Calendar.getInstance() |
| 51 | + cal.setTimeInMillis(self.native_instance.getSelection()) |
| 52 | + return cal.get(Calendar.YEAR), cal.get(Calendar.MONTH), cal.get(Calendar.DAY_OF_MONTH) |
| 53 | + |
| 54 | +else: |
| 55 | + # ======================================== |
| 56 | + # iOS class |
| 57 | + # ======================================== |
| 58 | + |
| 59 | + from rubicon.objc import ObjCClass |
| 60 | + from datetime import datetime |
| 61 | + |
| 62 | + class MaterialDatePicker(MaterialDatePickerBase, ViewBase): |
| 63 | + def __init__(self, year: int = 0, month: int = 0, day: int = 0) -> None: |
| 64 | + super().__init__() |
| 65 | + self.native_class = ObjCClass("UIDatePicker") |
| 66 | + self.native_instance = self.native_class.alloc().init() |
| 67 | + self.set_date(year, month, day) |
| 68 | + |
| 69 | + def set_date(self, year: int, month: int, day: int) -> None: |
| 70 | + date = datetime(year, month, day) |
| 71 | + self.native_instance.setDate_(date) |
| 72 | + |
| 73 | + def get_date(self) -> tuple: |
| 74 | + date = self.native_instance.date() |
| 75 | + return date.year, date.month, date.day |
0 commit comments