web3-wizzard-lib 1.2.1__tar.gz → 1.3.0__tar.gz

{web3_wizzard_lib-1.2.1 → web3_wizzard_lib-1.3.0}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.2
 Name: web3_wizzard_lib
-Version: 1.2.1
+Version: 1.3.0
 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==10.0.9
+Requires-Dist: sybil-engine==10.1.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: description-content-type

web3_wizzard_lib-1.3.0/requirements.txt

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

{web3_wizzard_lib-1.2.1 → web3_wizzard_lib-1.3.0}/setup.py

@@ -5,7 +5,7 @@ with open("requirements.txt", "r", encoding="utf-8") as file:
 
 setup(
     name='web3_wizzard_lib',
-    version='1.2.1',
+    version='1.3.0',
     py_modules=['web3_wizzard_lib'],
     packages=find_packages(),
     package_data={

{web3_wizzard_lib-1.2.1 → web3_wizzard_lib-1.3.0}/web3_wizzard_lib/core/modules/airdrop_printer.py

@@ -2,9 +2,10 @@ import requests
 from loguru import logger
 from sybil_engine.data.networks import get_chain_instance
 from sybil_engine.module.module import Module
-from sybil_engine.utils.statistic_utils import get_statistic_writer, statistic_date_string
 from web3 import Web3
 
+from web3_wizzard_lib.core.utils.statistic_utils import get_statistic_writer, statistic_date_string
+
 
 class AirdropPrinter(Module):
     module_name = 'AIRDROP_PRINTER'

{web3_wizzard_lib-1.2.1 → web3_wizzard_lib-1.3.0}/web3_wizzard_lib/core/modules/debank_checker.py

@@ -1,7 +1,8 @@
 import requests
 from sybil_engine.module.module import Module
 from sybil_engine.utils.accumulator import add_accumulator, get_value
-from sybil_engine.utils.statistic_utils import statistic_date_string, get_statistic_writer
+
+from web3_wizzard_lib.core.utils.statistic_utils import statistic_date_string, get_statistic_writer
 
 TOTAL_USD = "TotalUSD"
 DEBANK_ACC_NUM = "DEBANK_ACC_NUM"

{web3_wizzard_lib-1.2.1 → web3_wizzard_lib-1.3.0}/web3_wizzard_lib/core/modules/linea_appeal.py

@@ -4,11 +4,11 @@ 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 get_ai_chat
 
 from web3_wizzard_lib.core.utils.module_memory import get_by_key, accumulate_by_key, remove_key, add_value
+from web3_wizzard_lib.core.utils.statistic_utils import get_statistic_writer, statistic_date_string
 
 APPEAL_ACCOUNTS = "APPEAL_ACCOUNTS"
 APPEAL_ACCOUNTS_AMOUNT = "APPEAL_ACCOUNTS_AMOUNT"

{web3_wizzard_lib-1.2.1 → web3_wizzard_lib-1.3.0}/web3_wizzard_lib/core/modules/linea_poh_lxp.py

@@ -5,9 +5,10 @@ from sybil_engine.domain.balance.balance import Erc20Balance
 from sybil_engine.domain.balance.tokens import Erc20Token
 from sybil_engine.module.module import Module
 from sybil_engine.utils.accumulator import add_accumulator_balance, add_accumulator, get_value
-from sybil_engine.utils.statistic_utils import statistic_date_string, get_statistic_writer
 from sybil_engine.utils.web3_utils import init_web3
 
+from web3_wizzard_lib.core.utils.statistic_utils import statistic_date_string, get_statistic_writer
+
 
 class LineaPOHLXP(Module):
     module_name = 'LINEA_POH_LXP'

web3_wizzard_lib-1.3.0/web3_wizzard_lib/core/utils/statistic_utils.py

@@ -0,0 +1,152 @@
+import csv
+import datetime
+import os
+from google.oauth2.service_account import Credentials
+from googleapiclient.discovery import build
+from loguru import logger
+
+from sybil_engine.utils.telegram import get_config
+
+statistic_date_string = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+
+
+class StatisticsWriter:
+    def init_if_required(self, job_name, header):
+        pass
+
+    def write_row(self, job_name, row):
+        pass
+
+
+class CsvStatisticsWriter(StatisticsWriter):
+    def init_if_required(self, job_name, header):
+        output_file = self._get_file_name(job_name)
+        file_exists = os.path.isfile(output_file)
+
+        rows_to_write = []
+
+        if not file_exists or os.path.getsize(output_file) == 0:
+            rows_to_write.append(header)
+
+        with open(output_file, mode='a', newline='') as file:
+            writer = csv.writer(file)
+            writer.writerows(rows_to_write)
+
+    def write_row(self, job_name, row):
+        rows_to_write = []
+        output_file = self._get_file_name(job_name)
+
+        rows_to_write.append(row)
+
+        with open(output_file, mode='a', newline='') as file:
+            writer = csv.writer(file)
+            writer.writerows(rows_to_write)
+
+    def _get_file_name(self, job_name):
+        return f'data/csv/{job_name}.csv'
+
+
+class GoogleStatisticsWriter(StatisticsWriter):
+    def __init__(self, sheet):
+        self.SERVICE_ACCOUNT_FILE = 'data/service-accounts.json'
+        self.SCOPES = ['https://www.googleapis.com/auth/spreadsheets']
+        self.credentials = Credentials.from_service_account_file(self.SERVICE_ACCOUNT_FILE, scopes=self.SCOPES)
+        self.service = build('sheets', 'v4', credentials=self.credentials)
+
+        self.sheet = sheet
+
+    def copy_sheet(self, source_sheet_name, destination_sheet_name):
+        # Get the source sheet ID
+        source_sheet = self.service.spreadsheets().get(spreadsheetId=self.sheet).execute()
+        source_sheet_id = next(
+            (sheet['properties']['sheetId'] for sheet in source_sheet['sheets'] if
+             sheet['properties']['title'] == source_sheet_name),
+            None
+        )
+
+        if source_sheet_id is None:
+            raise ValueError(f"Sheet '{source_sheet_name}' not found.")
+
+        # Create the request body for copying the sheet
+        request_body = {
+            'destinationSpreadsheetId': self.sheet
+        }
+
+        # Copy the sheet to the same spreadsheet with the specified name
+        result = self.service.spreadsheets().sheets().copyTo(
+            spreadsheetId=self.sheet,
+            sheetId=source_sheet_id,
+            body=request_body
+        ).execute()
+
+        # Rename the copied sheet
+        body = {
+            'requests': [
+                {
+                    'updateSheetProperties': {
+                        'properties': {
+                            'sheetId': result['sheetId'],
+                            'title': destination_sheet_name
+                        },
+                        'fields': 'title'
+                    }
+                }
+            ]
+        }
+        self.service.spreadsheets().batchUpdate(spreadsheetId=self.sheet, body=body).execute()
+
+        logger.info(f"Sheet copied. New sheet name: {destination_sheet_name}")
+
+    def init_if_required(self, sheet_name, header):
+        if not self._sheet_exists(sheet_name):
+            self._create_new_sheet(sheet_name)
+            self.write_row(sheet_name, header)
+
+    def _sheet_exists(self, sheet_name):
+        try:
+            self.service.spreadsheets().values().get(spreadsheetId=self.sheet, range=sheet_name).execute()
+            logger.info('Sheet exists.')
+            return True
+        except Exception:
+            return False
+
+    def _create_new_sheet(self, sheet_name):
+        body = {'requests': [{'addSheet': {'properties': {'title': sheet_name}}}]}
+        self.service.spreadsheets().batchUpdate(spreadsheetId=self.sheet, body=body).execute()
+        logger.info(f'Sheet "{sheet_name}" created.')
+
+    def write_cell(self, sheet_name, cell_reference, value):
+        range_ = f"{sheet_name}!{cell_reference}"
+        body = {'values': [[value]]}
+        self.service.spreadsheets().values().update(
+            spreadsheetId=self.sheet, range=range_, valueInputOption='USER_ENTERED', body=body
+        ).execute()
+        logger.info(f"Value written to cell {cell_reference} in sheet {sheet_name}: {value}")
+
+    def write_row(self, job_name, row):
+        range_ = f"{job_name}!A1"
+        rows_read = self.service.spreadsheets().values().get(spreadsheetId=self.sheet, range=range_).execute().get(
+            'values', [])
+        body = {'values': [row]}
+        range_to_write = f"{job_name}!A{len(rows_read) + 1}"
+        self.service.spreadsheets().values().append(
+            spreadsheetId=self.sheet, range=range_to_write, valueInputOption='USER_ENTERED', body=body
+        ).execute()
+        logger.info(f'Row written: {row}')
+
+    def get_leftmost_sheet(self):
+        """
+        Returns the name of the leftmost sheet in the spreadsheet.
+        The leftmost sheet is the first sheet in the sheets collection.
+        """
+        spreadsheet = self.service.spreadsheets().get(spreadsheetId=self.sheet).execute()
+        if 'sheets' in spreadsheet and len(spreadsheet['sheets']) > 0:
+            return spreadsheet['sheets'][0]['properties']['title']
+        return None
+
+
+def get_statistic_writer():
+    if get_config("STATISTICS_MODE") == "CSV":
+        return CsvStatisticsWriter()
+    else:
+        return GoogleStatisticsWriter(get_config("SPREADSHEET_ID"))

{web3_wizzard_lib-1.2.1 → web3_wizzard_lib-1.3.0}/web3_wizzard_lib.egg-info/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.2
 Name: web3_wizzard_lib
-Version: 1.2.1
+Version: 1.3.0
 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==10.0.9
+Requires-Dist: sybil-engine==10.1.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: description-content-type

{web3_wizzard_lib-1.2.1 → web3_wizzard_lib-1.3.0}/web3_wizzard_lib.egg-info/SOURCES.txt

@@ -267,6 +267,7 @@ web3_wizzard_lib/core/utils/__init__.py
 web3_wizzard_lib/core/utils/ai_utils.py
 web3_wizzard_lib/core/utils/benchmark_utils.py
 web3_wizzard_lib/core/utils/module_memory.py
+web3_wizzard_lib/core/utils/statistic_utils.py
 web3_wizzard_lib/core/utils/sub_module.py
 web3_wizzard_lib/resources/banner.txt
 web3_wizzard_lib/resources/linea_appeal.txt

web3_wizzard_lib-1.3.0/web3_wizzard_lib.egg-info/requires.txt

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

web3_wizzard_lib-1.2.1/requirements.txt

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

web3_wizzard_lib-1.2.1/web3_wizzard_lib.egg-info/requires.txt

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