web3-wizzard-lib 0.0.3__py3-none-any.whl → 0.1.2__py3-none-any.whl

web3_wizzard_lib/core/modules/linea_appeal.py

@@ -1,9 +1,18 @@
+import random
 import webbrowser
 
 from loguru import logger
 from sybil_engine.module.module import Module
+from sybil_engine.utils.accumulator import add_accumulator, get_value
+from sybil_engine.utils.statistic_utils import get_statistic_writer, statistic_date_string
 
-from web3_wizzard_lib.core.utils.ai_utils import ChatGPT
+from web3_wizzard_lib.core.utils.ai_utils import get_ai_chat
+from sybil_engine.utils.telegram import add_config
+
+from web3_wizzard_lib.core.utils.module_memory import get_by_key, accumulate_by_key, remove_key
+
+APPEAL_ACCOUNTS = "APPEAL_ACCOUNTS"
+APPEAL_ACCOUNTS_AMOUNT = "APPEAL_ACCOUNTS_AMOUNT"
 
 
 class LineaAppeal(Module):
@@ -14,24 +23,65 @@ class LineaAppeal(Module):
     with open("resources/linea_appeal.txt") as f:
         linea_appeal_reason = f.read()
 
-    def execute(self, account, token, accounts, statistic_write="GOOGLE"):
-        chat_gpt = ChatGPT(token)
+    def execute(self, token, accounts, statistic_write, ai_type, account):
+        add_accumulator("Acc Num", 1)
 
-        reason = chat_gpt.ask(self.linea_appeal_reason)
+        statistics = get_statistic_writer()
+        statistics.init_if_required(
+            f"linea_appeal_{statistic_date_string}",
+            ["#", "MainAddress", "GPT Answer"]
+        )
 
+        chat_gpt = get_ai_chat(ai_type, token)
+        reason = chat_gpt.ask(self.linea_appeal_reason)
         logger.info(reason)
 
+        statistics.write_row(
+            statistic_date_string,
+            [account.app_id, account.address, reason]
+        )
+
+        if get_by_key(APPEAL_ACCOUNTS_AMOUNT) is None:
+            add_config(
+                APPEAL_ACCOUNTS_AMOUNT,
+                random.randint(accounts['from'], accounts['to'])
+            )
+
+        accumulate_by_key(
+            APPEAL_ACCOUNTS, {
+                "address": account.address,
+                "reason": reason,
+            }
+        )
+
+        logger.info(f"Acc: {get_value("Acc Num")}")
+        logger.info(f"Acc Amount: {get_value("Acc Amount")}")
+
+        if (get_by_key(APPEAL_ACCOUNTS_AMOUNT) == len(get_by_key(APPEAL_ACCOUNTS))
+                or get_value("Acc Num") == get_value("Acc Amount")):
+            wallets = get_by_key(APPEAL_ACCOUNTS)
+            address_list = [wallet["address"] for wallet in wallets]
+            address_list.remove(account.address)
+            formatted_string = "\n".join(f"{wallet['address']}\n{wallet['reason']}" for wallet in wallets)
+            self.open_appeal_form(account, "\n".join(address_list), formatted_string)
+            remove_key(APPEAL_ACCOUNTS)
+            remove_key(APPEAL_ACCOUNTS_AMOUNT)
+
+    def open_appeal_form(self, account, address_list, formatted_string):
         payload = {
             "entry.1292139045": account.address,
-            "entry.1099559693": "",
-            "entry.1296389817": reason['choices'][0]['message']['content']
+            "entry.1099559693": address_list.replace("\n", "%0A"),
+            "entry.1296389817": formatted_string.replace("\n", "%0A")
         }
-
         query_string = "&".join(f"{key}={value}" for key, value in payload.items())
         pre_filled_url = f"{self.base_url}?{query_string}"
-
         print(f"Opening form: {pre_filled_url}")
         webbrowser.open(pre_filled_url)
 
     def parse_params(self, module_params):
-        return module_params['token'], module_params['accounts']
+        return (
+            module_params['ai_token'],
+            module_params['accounts'],
+            module_params['write_mode'],
+            module_params['ai_type']
+        )

web3_wizzard_lib/core/utils/ai_utils.py

@@ -1,5 +1,7 @@
 import requests
 
+from web3_wizzard_lib.core.utils.benchmark_utils import benchmark
+
 
 class AIChat:
     def ask(self, question):
@@ -10,6 +12,7 @@ class ChatGPT:
     def __init__(self, token):
         self.token = token
 
+    @benchmark
     def ask(self, question):
         url = "https://api.openai.com/v1/chat/completions"
         headers = {
@@ -24,4 +27,19 @@ class ChatGPT:
 
         response = requests.post(url, headers=headers, json=data)
 
-        return response.json()
+        return response.json()['choices'][0]['message']['content']
+
+
+class MockAIChat:
+    def __init__(self, token):
+        self.token = token
+
+    def ask(self, question):
+        return f"ANSWER {question}"
+
+
+def get_ai_chat(ai_config, token):
+    if ai_config == 'CHAT_GPT':
+        return ChatGPT(token)
+    else:
+        return MockAIChat(token)

web3_wizzard_lib/core/utils/benchmark_utils.py

@@ -0,0 +1,12 @@
+import time
+
+
+def benchmark(func):
+    def wrap(*args, **kwargs):
+        start = time.perf_counter()
+        result = func(*args, **kwargs)
+        end = time.perf_counter()
+        print(f"{func.__name__} executed in {end - start:.4f} seconds")
+        return result
+
+    return wrap

web3_wizzard_lib/core/utils/module_memory.py

@@ -0,0 +1,26 @@
+memory = {
+
+}
+
+
+def add_memory_list(key, value):
+    memory[key] = value
+
+
+def accumulate_by_key(key, value):
+    if key not in memory:
+        memory[key] = []
+
+    memory[key].append(value)
+    # if isinstance(value, map):
+    #     (list(value))
+    # else:
+    #     memory[key] += value
+
+
+def remove_key(key):
+    del memory[key]
+
+
+def get_by_key(key):
+    return memory.get(key)

web3_wizzard_lib-0.1.2.data/data/requirements.txt

@@ -0,0 +1 @@
+sybil-engine==10.0.2

{web3_wizzard_lib-0.0.3.dist-info → web3_wizzard_lib-0.1.2.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.2
 Name: web3_wizzard_lib
-Version: 0.0.3
+Version: 0.1.2
 Summary: Engine for web3 smart contracts automatization.
 Home-page: https://github.com/Indeoo/web3-wizzard-lib/
 Author: Indeoo
 Author-email: indeooars@gmail.com
 Description-Content-Type: text/markdown
-Requires-Dist: sybil-engine==9.9.0
+Requires-Dist: sybil-engine==10.0.2
 Dynamic: author
 Dynamic: author-email
 Dynamic: description-content-type

{web3_wizzard_lib-0.0.3.dist-info → web3_wizzard_lib-0.1.2.dist-info}/RECORD

@@ -105,7 +105,7 @@ web3_wizzard_lib/core/modules/erc20_balance.py,sha256=4mxd6QpOgnHHqUB0u6ya1Kh_xw
 web3_wizzard_lib/core/modules/intract_claim.py,sha256=TyWtlZ6Rlx8ukG2elc4Lcr_ImF1K2RL2lz_EkhNfJBc,3744
 web3_wizzard_lib/core/modules/layer_2_20.py,sha256=1MvidKQBS5GhBzH6yAmjTf1VHnM96_Wb0sGF24TTXJo,2475
 web3_wizzard_lib/core/modules/lending_module.py,sha256=Sy0P-Ojt3e1JVMUCqpG06-DsiuTCvvHilgSj_QJPCwg,5517
-web3_wizzard_lib/core/modules/linea_appeal.py,sha256=600wItAys3yzDYMh8QdGnIzvNRxvDEWQle8FTpWcVtc,1189
+web3_wizzard_lib/core/modules/linea_appeal.py,sha256=KPkPt97pko_9ENVverdtm-v1J-mHHmWsR0mjTdoHUXU,3267
 web3_wizzard_lib/core/modules/linea_poh_lxp.py,sha256=keipTIhT3SrOESNJdIJesHBOOM-NKVzMGqKEACEKIvQ,2320
 web3_wizzard_lib/core/modules/liquidity_pool.py,sha256=vu0vBoXomvAJFa10n9woo66v62hQbEO726M4lkiLWNE,2181
 web3_wizzard_lib/core/modules/merkly_refuel.py,sha256=whQD6CplGt-hQxFN7lg40F6SmFU5niOO529lKMj0dyA,3488
@@ -252,10 +252,12 @@ web3_wizzard_lib/core/modules/swap/woofi.py,sha256=X1I6T4EVYpPVOdYuO_RALFubslCTm
 web3_wizzard_lib/core/modules/swap/xy_finance.py,sha256=PO-dcFhPzr3D5a9-D31vPCgqcs-_SUf1YXCE85Bjehw,2058
 web3_wizzard_lib/core/modules/swap/zebra.py,sha256=AnoBSRkh1YFyIbbL87JN9JwMTKnyGt5XY5K_7C7DcyM,1423
 web3_wizzard_lib/core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-web3_wizzard_lib/core/utils/ai_utils.py,sha256=5FlQhbMPRcEkSVkfDxx81VvutSsyYhBbxc4P-1Qy7yE,626
+web3_wizzard_lib/core/utils/ai_utils.py,sha256=v1aVEAQz9IoI1B9qdfeDUohoMrHr8EBFKBFbqM8Ddng,1030
+web3_wizzard_lib/core/utils/benchmark_utils.py,sha256=3YBPXseWJb_BpBJtWWU6Dc5kydyhX4YWSvEWj4v6ekk,286
+web3_wizzard_lib/core/utils/module_memory.py,sha256=-EoEG5KvmWxGervESzCw0htjraFvNw6Fck7_aMwU4NE,385
 web3_wizzard_lib/core/utils/sub_module.py,sha256=r7C89nhlOUAYtCI92JINapS_-5hUUYX7YnY9hQLgHKg,117
-web3_wizzard_lib-0.0.3.data/data/requirements.txt,sha256=H-xf3U_08LGLn-9XSjY5SNImgm9ZGl5DwBcy5-lL-pQ,19
-web3_wizzard_lib-0.0.3.dist-info/METADATA,sha256=nEqfuZ35me3ZhpAj_brCTOvUVxdaVarZ4wrnMlongLA,427
-web3_wizzard_lib-0.0.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-web3_wizzard_lib-0.0.3.dist-info/top_level.txt,sha256=8dD8S5HQo4dKtxogttWY_sh4n3yCVy0MwiTelYp6kug,28
-web3_wizzard_lib-0.0.3.dist-info/RECORD,,
+web3_wizzard_lib-0.1.2.data/data/requirements.txt,sha256=YhDe0eUumhIfPZauGghmAKIlbFEO4wyxP2iEMUC6Z28,20
+web3_wizzard_lib-0.1.2.dist-info/METADATA,sha256=_aoL2rNEBLo0A3TPUCoDNNXMtwG9o3nYhF0YYca3baE,428
+web3_wizzard_lib-0.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+web3_wizzard_lib-0.1.2.dist-info/top_level.txt,sha256=8dD8S5HQo4dKtxogttWY_sh4n3yCVy0MwiTelYp6kug,28
+web3_wizzard_lib-0.1.2.dist-info/RECORD,,

web3_wizzard_lib-0.0.3.data/data/requirements.txt

@@ -1 +0,0 @@
-sybil-engine==9.9.0