Upload code

tasks/item/assets/assets_item_data.py

@@ -0,0 +1,25 @@
+from module.base.button import Button, ButtonWrapper
+
+# This file was auto-generated, do not modify it manually. To generate:
+# ``` python -m dev_tools.button_extract ```
+
+OCR_AP = ButtonWrapper(
+    name='OCR_AP',
+    share=Button(
+        file='./assets/share/item/data/OCR_AP.png',
+        area=(560, 11, 667, 37),
+        search=(540, 0, 687, 57),
+        color=(211, 216, 219),
+        button=(560, 11, 667, 37),
+    ),
+)
+OCR_DATA = ButtonWrapper(
+    name='OCR_DATA',
+    share=Button(
+        file='./assets/share/item/data/OCR_DATA.png',
+        area=(768, 12, 1072, 37),
+        search=(748, 0, 1092, 57),
+        color=(212, 220, 224),
+        button=(768, 12, 1072, 37),
+    ),
+)

tasks/item/data_update.py

@@ -0,0 +1,47 @@
+from module.base.timer import Timer
+from module.logger import logger
+from module.ocr.ocr import Digit, DigitCounter
+from tasks.base.page import page_work
+from tasks.base.ui import UI
+from tasks.item.assets.assets_item_data import *
+
+
+class DataUpdate(UI):
+    def _get_data(self):
+        """
+        Page:
+            in: page_work
+        """
+        ap = DigitCounter(OCR_AP).ocr_single_line(self.device.image)
+        # Data for Credit and Pyroxene
+        ocr = Digit(OCR_DATA)
+        timeout = Timer(2, count=6).start()
+        while 1:
+            data = ocr.detect_and_ocr(self.device.image)
+            if len(data) != 2:
+                data = [data[0], data[-1]]
+            logger.attr('Data', data)
+            credit, pyroxene = [int(''.join([v for v in d.ocr_text if v.isdigit()])) for d in data]
+            if credit > 0 or pyroxene > 0:
+                break
+
+            logger.warning(f'Invalid credit and pyroxene: {data}')
+            if timeout.reached():
+                logger.warning('Get data timeout')
+                break
+
+        logger.attr('Credit', credit)
+        logger.attr('Pyroxene', pyroxene)
+        with self.config.multi_set():
+            self.config.stored.AP.set(ap[0], ap[2])
+            self.config.stored.Credit.value = credit
+            self.config.stored.Pyroxene.value = pyroxene
+
+        return ap, credit, pyroxene
+
+    def run(self):
+        self.ui_ensure(page_work, acquire_lang_checked=False)
+
+        with self.config.multi_set():
+            self._get_data()
+            self.config.task_delay(server_update=True)