Apply current workspace changes

2026-04-22 01:52:21 +08:00
parent cf34b9a8ed
commit cb518b431f
449 changed files with 92252 additions and 321 deletions
@@ -0,0 +1,83 @@
+import time
+from pathlib import Path
+from typing import Optional, Union
+
+import numpy as np
+import requests
+from tqdm.auto import tqdm
+
+from .datasets import SarDataset
+
+
+class Submitter:
+    def __init__(
+        self,
+        dataset: SarDataset,
+        email: str,
+        sleep_time_range: tuple[int, int] = (60 / 2, 60 * 5),
+        gacos_url="http://www.gacos.net/M/action_page.php",
+    ) -> None:
+        """Initialize Submitter class
+
+        Parameters
+        ----------
+        dataset : SarDataset
+            The SarDataset object.
+        email : str
+            The email address to submit to gacos.
+        sleep_time_range : tuple[int, int], optional
+            The range of sleep time in seconds. Default is (60, 60 * 5).
+        gacos_url : str, optional
+            The url of gacos website. Default is "http://www.gacos.net/M/action_page.php".
+        """
+        self.dataset = dataset
+        self.email = email
+        self.sleep_time_range = sleep_time_range
+        self.gacos_url = gacos_url
+
+        self._failed = []
+        self._succeed = []
+
+    def _post_data(self, data):
+        """Post data to gacos website."""
+        r = requests.post(self.gacos_url, data=data)
+        return "Thanks for using GACOS!" in r.text
+
+    def post_requests(self):
+        # post gacos info to website
+        datetime_patches = self.dataset.gen_datetime_patches()
+        for _key, _dates in tqdm(
+            datetime_patches.items(),
+            desc="submitting times",
+            unit="times",
+        ):
+            try:
+                for _dt in tqdm(_dates, desc="submitting dates", unit="dates"):
+                    post_data = self.dataset.gen_post_data(
+                        _dt, _key.split(":"), self.email
+                    )
+                    status_ok = self._post_data(post_data)
+                    if status_ok:
+                        self._succeed.append(post_data)
+                        tqdm.write(f">>> succeed post: {post_data}")
+                    else:
+                        self._failed.append(post_data)
+                        tqdm.write(f">>> failed post: {post_data}")
+
+                    # wait to avoid be rejected
+                    sleep_time = np.random.randint(*self.sleep_time_range)
+                    tqdm.write(f"    sleeping for {sleep_time} seconds...")
+                    time.sleep(sleep_time)
+            except:
+                self._failed.append(post_data)
+                tqdm.write(f">>> failed post: {post_data}")
+
+    @property
+    def failed(self):
+        """A list of failed post data."""
+        return self._failed
+
+    @property
+    def succeed(self):
+        """A list of succeed post data."""
+        return self._succeed