optimuslib 0.0.36__tar.gz → 0.0.38__tar.gz

{optimuslib-0.0.36 → optimuslib-0.0.38}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: optimuslib
-Version: 0.0.36
+Version: 0.0.38
 Summary: Function Library for mostly used codes
 Author: Shomi Nanwani
 Requires-Python: >=3.9,<4.0

{optimuslib-0.0.36 → optimuslib-0.0.38}/optimuslib/optimuslib.py

@@ -149,12 +149,12 @@ osType, osVersion, oshostname = checkOS()
 if osType == 'Windows':
     proxyhost = 'http://127.0.0.1:8080'
     proxyhost = None
-    log.setLevel(logging.INFO)    
-    file.setLevel(logging.INFO)
-    stream.setLevel(logging.INFO)
-    # log.setLevel(logging.DEBUG)
-    # file.setLevel(logging.DEBUG)
-    # stream.setLevel(logging.DEBUG)
+    # log.setLevel(logging.INFO)    
+    # file.setLevel(logging.INFO)
+    # stream.setLevel(logging.INFO)
+    log.setLevel(logging.DEBUG)
+    file.setLevel(logging.DEBUG)
+    stream.setLevel(logging.DEBUG)
 else:
     proxyhost = None
     # log.setLevel(logging.INFO)    
@@ -1034,14 +1034,18 @@ def append_array_to_csv(file_path, data, headers=None):
     else:
         df.to_csv(file_path, mode='a', index=False, header=False)
 
+# def write_array_to_csv(file_path, data, headers=None):
+#     df = pd.DataFrame([data])
+#     if headers is not None:
+#         df.columns = headers
+#     if not os.path.exists(file_path) or os.stat(file_path).st_size == 0:
+#         df.to_csv(file_path, mode='w', index=False, header=headers is not None)
+#     else:
+#         df.to_csv(file_path, mode='w', index=False, header=False)
 def write_array_to_csv(file_path, data, headers=None):
-    df = pd.DataFrame([data])
-    if headers is not None:
-        df.columns = headers
-    if not os.path.exists(file_path) or os.stat(file_path).st_size == 0:
-        df.to_csv(file_path, mode='w', index=False, header=headers is not None)
-    else:
-        df.to_csv(file_path, mode='w', index=False, header=False)
+    with open(file_path, 'w', newline='') as file:
+        writer = csv.writer(file)
+        writer.writerows(data)
 
 def read_csv_to_arrays(file_path):
     if not os.path.exists(file_path) or os.stat(file_path).st_size == 0:
@@ -1192,7 +1196,8 @@ def printAndSend(botToken,chatId,output):
 
 def logAndSend(botToken,chatId,output):
     log.info(output)
-    sendTelegramBotNotification(botToken,chatId,output)
+    # sendTelegramBotNotification(botToken,chatId,output)
+    sendDiscordBotNotification(botToken,chatId,output)
 
 ########## format a number as per indian currency 
 def formatCurrrencyIndian(number):
@@ -1218,7 +1223,162 @@ def formatCurrrencyIndian(number):
     
     return formatted_number
 
+########### DISCORD BOT NOTIFICATION - BOT DETAILS ##############
+# def getDiscordBotInfo(botToken):
+#     log.info('Fetching BOT Information')
+#     url = 'https://api.telegram.org/bot'+botToken+'/getMe'
+#     response = sendRequest('get', url, proxyhost=proxyhost)
+#     responseJson = response.result().json()
+#     if responseJson['ok']==True:
+#         if responseJson['result']:
+#             id = responseJson['result']['id']
+#             is_bot = responseJson['result']['is_bot']
+#             first_name = responseJson['result']['first_name']
+#             username = responseJson['result']['username']
+#             can_join_groups = responseJson['result']['can_join_groups']
+#             can_read_all_group_messages = responseJson['result']['can_read_all_group_messages']
+#             supports_inline_queries = responseJson['result']['supports_inline_queries']
+#             getVarInfo('id',id)
+#             getVarInfo('is_bot',is_bot)
+#             getVarInfo('first_name',first_name)
+#             getVarInfo('username',username)
+#             getVarInfo('can_join_groups',can_join_groups)
+#             getVarInfo('can_read_all_group_messages',can_read_all_group_messages)
+#             getVarInfo('supports_inline_queries',supports_inline_queries)
+#             return True
+#         else:
+#             return False
+#     else:
+#         getVarInfo('responseJson',responseJson)
+#         return False
+
+########### DISCORD BOT NOTIFICATION - FETCH RECEIVED MESSAGES ##############
+def getDiscordBotUpdates(botToken, channelId, messageCount=1):
+    log.info('Checking new messages...')
+    url = 'https://discord.com/api/v10/channels/'+channelId+'/messages?limit='+str(messageCount)
+    headers = {
+        'Authorization': 'Bot ' + botToken,
+        'Content-Type': 'application/json'
+    }
+    response = sendRequest('get', url, headers=headers, proxyhost=proxyhost)
+    # response = sendRequest('get', url, proxyhost=proxyhost)
+    # try:
+    if response.result().status_code == 200:
+        responseJson = response.result().json()
+        for msg in responseJson:
+            author = msg['author']
+            content = msg['content']
+            timestamp = msg['timestamp']
+            id = msg['id']
+            username = author['username']
+            # print(f"[{author['username']}] {content}")
+
+            getVarInfo('id',id)
+            getVarInfo('timestamp',timestamp)
+            getVarInfo('username',username)
+            getVarInfo('content',content)
+
+            return id, timestamp, username, content
+    #         else:
+    #             log.info('\tNo Data in Result')
+    #             return 0, False, False, False, False
+    #     else:
+    #         getVarInfo('responseJson',responseJson)
+    #         return 0, False, False, False, False
+    # except (KeyError, TypeError, ValueError) as e:
+    #     log.info('Exception Occured 202502042248: %s', e)
+    #     getVarInfo('responseJson',responseJson)
+    #     return 0, False, False, False, False
+    
+
+# def clearDiscordBotUpdates(botToken, update_id=0):
+#     log.info('Checking new messages...')
+#     url = 'https://api.telegram.org/bot'+botToken+'/getUpdates?offset='+str(update_id+1)
+#     response = sendRequest('get', url, proxyhost=proxyhost)
+#     try:
+#         responseJson = response.result().json()
+#         if responseJson['ok']==True:
+#             if responseJson['result']:
+#                 update_id = responseJson['result'][0]['update_id']
+#                 clearTelegramBotUpdates(botToken, update_id)
+#                 return True
+#         else:
+#             getVarInfo('responseJson',responseJson)
+#             return False
+#     except (KeyError, TypeError, ValueError) as e:
+#         log.info('Exception Occured: %s', e)
+#         getVarInfo('responseJson',responseJson)
+#         return False
+
+def fetchOTPDiscord(botToken, otpFetchWaitTime, otpChannelId):
+    # 2FA - autofetch via telegram
+    log.info('Fetching 2FA request with TOTP')
+    twofa_value = 0
+    retryCount = 0
+    while retryCount<60:
+        log.info('Sleeping for %d seconds', otpFetchWaitTime)
+        time.sleep(otpFetchWaitTime)
+        try:
+            id, timestamp, username, content = getDiscordBotUpdates(botToken, otpChannelId, messageCount=1)
+            log.info('update_id: %s, text: %s', id, content)
+            twofa_value = content[-6:]
+            getVarInfo('twofa_value',twofa_value)
+            break
+        except ValueError as e:
+            retryCount += 1
+            log.exception('Retrying %d since ValueError: %s',retryCount, e)
+        # print(update_id, text)
+    return twofa_value
+
+########### DISCORD BOT NOTIFICATION - SEND ##############
+
+# function to validate if length is > 4096
+def sendDiscordBotNotification(botToken,channelId,message, parse_mode=''):
+    msgs = [message[i:i + 4096] for i in range(0, len(message), 4096)]
+    for text in msgs:
+         sendDiscordBotNotificationMain(botToken,channelId,text, parse_mode)
+            
+def sendDiscordBotNotificationMain(botToken,channelId,message, parse_mode=''):
+    log.info('Sending Message via BOT')
+    
+    url = f'https://discord.com/api/v10/channels/{channelId}/messages'
+    data = {'content': message}
+    headers = {'Authorization': 'Bot ' + botToken}
+    # proxyhost= 'http://127.0.0.1:8080'
+    response = sendRequest('post', url, headers=headers, json=data, proxyhost=proxyhost)
+    if response.result().status_code == 200:
+        return True
+    else:
+        responseJson = response.result().json()
+        log.info('Discord Bot Message Send Failure Error Description: %s', responseJson['description'])
+        getVarInfo('responseJson',responseJson)
+        return False
+        
+# send telegram document
+def sendDiscordBotFile(botToken,channelId,filePath):
+    documentFile = open(filePath, 'rb')
+    url = f'https://discord.com/api/v10/channels/{channelId}/messages'
+    files = {'file': documentFile}
+    headers = {'Authorization': 'Bot ' + botToken}
+    response = sendRequest('post', url, headers=headers, files=files, proxyhost=proxyhost)
+    # response = sendRequest('post', url, params=params, files=files, proxyhost=proxyhost)
+    responseJson = response.result().json()
+    if response.result().status_code == 200:
+        if responseJson['type']==0:
+            return True
+    else:
+        log.info('Discord Bot Message Send Failure Error Description: %s', responseJson['description'])
+        getVarInfo('responseJson',responseJson)
+        return False
+   
+# if(optimuslib.sendTelegramNotification(botToken, chatId,output)):
+    # print('Telegram Notification Sent.')
+
 
+# getDiscordBotUpdates(botToken, channelId, messageCount)
+# fetchOTPDiscord(botToken, otpFetchWaitTime, otpChannelId)
+# sendDiscordBotNotification(botToken,channelId,message)
+# sendDiscordBotFile(botToken,channelId,filePath)
 ################################################
 log.info('[+] Optimuslib Import Sucessfull.')
 

{optimuslib-0.0.36 → optimuslib-0.0.38}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "optimuslib"
-version = "0.0.36"
+version = "0.0.38"
 description = "Function Library for mostly used codes"
 authors = ["Shomi Nanwani"]
 readme = "README.md"