Upload 4 files

WarOnline_Chat.py

@@ -52,12 +52,14 @@ def remove_non_english_russian_chars(s):
     return re.sub(pattern, '', s)
 
 def remove_extra_spaces(s):
+    # Removes extra whitespaces and unwanted characters
     s = re.sub(r"\s+", " ", s)  # replace all sequences of whitespace with a single space
     s = re.sub(r"\s+([.,-])", r"\1", s)  # remove spaces before period, dash or comma
     return(s)
 
 def getLastPage(thread_url=config.thread_url):
     # Returns the number of the last page
+    print('looking for the last page of the thread')
     page = 1  # Starting page
     lastPage = False
 
@@ -66,6 +68,7 @@ def getLastPage(thread_url=config.thread_url):
             page += 1
         else:
             lastPage = True
+    print('Last page of this thread is '+str(page))
     return page
 
 def login(username=config.username, password=config.password, thread_url=config.thread_url):
@@ -90,6 +93,8 @@ def login(username=config.username, password=config.password, thread_url=config.
     if 'Invalid login' in response.text:
         print('Login failed!')
         exit()
+    else:
+        print('Login successful')
 
 def post(message="", thread_url=config.thread_url, post_url=config.post_url, quoted_by="",quote_text="",quote_source="",img_url=""):
     #Post a message to the forum (with or without the quote
@@ -216,129 +221,133 @@ def getMessages(thread_url=config.thread_url, quotedUser="", startingPage=1):
 
     return allquotes
 
-# Core Engine of the Client
 def WarOnlineBot():
+    # Core Engine of the Client
 
-    lookUpPages = 5 # How many pages back to look in the thread
-    startingPage = getLastPage(thread_url=config.thread_url) - lookUpPages
-    if startingPage < 1:
-        startingPage = 1 # Starting page cannot be less than 1
-
-    login(username=config.username, password=config.password, thread_url=config.thread_url)
-    #print("logged in")
-
-    # All messages (with quotes) by ALL users:
-    allMessages = getMessages(thread_url=config.thread_url, quotedUser='', startingPage=startingPage)
-
-    # IDs of the quoted messages, replied by the bot:
-    messages_by_bot_IDs = []
-
-    # Initiate the direct messages
-    direct_messages = []
-
-    for msg in allMessages:
-        # Direct message to the bot
-        if msg['quotedID'].split(': ')[-1] == '0': #debug
-            direct_messages.append(msg)
-        # Set a list of replied messages IDs
-        if msg['messengerName'] == config.username: #message posted by the WarBot
-            messages_by_bot_IDs.append(msg['quotedID'].split(': ')[-1])
-    # remove empty and repeated elements
-    messages_by_bot_IDs = list(set([elem for elem in messages_by_bot_IDs if elem]))
-
-    # All messages (with quotes) sent _FOR_ the Bot:
-    messagesForBot = getMessages(thread_url=config.thread_url, quotedUser=config.username, startingPage=startingPage)
-
-    # Append the direct messages to the messagesForBot:
-    for msg in direct_messages:
-        messagesForBot.append(msg)
-
-    # IDs of the messages, quoting the bot:
-    messages_for_bot_IDs = []
-
-    for msg in messagesForBot:
-        # Set a list of posted message IDs
-        messages_for_bot_IDs.append(msg['messageID'].split('-')[-1])
-    # remove empty elements
-    messages_for_bot_IDs = [elem for elem in messages_for_bot_IDs if elem]
-
-    # Filter to leave just the unanswered messages IDs:
-    messages_for_bot_IDs = [ID for ID in messages_for_bot_IDs if ID not in messages_by_bot_IDs]
-
-
-    # Reply the unanswered messages:
-    for msg in messagesForBot:
-        if msg['messageID'].split('-')[-1] in messages_for_bot_IDs:
-
-            originalQuote = msg['reply']
-            if originalQuote == "": # Just images, no text
-                continue
-            else:
-                quote = remove_non_english_russian_chars(msg['reply'])
-                quote = remove_extra_spaces(quote)
-
-            message = "" #Initiating the reply message by Bot
-            previous_dialogue = "" #Initiating the previous dialogue
-
-            print('Quote: ', originalQuote)
-
-            # Init Connection
-            db = conversationDB.DataBase()
-
-            if msg['quotedID'].split(': ')[-1] != '0':  # It is dialogue. Look-up for the previous quotes
-
-                # Get the previous dialogue from the database
-                dbmessages = db.getmessages(msg['messengerName'])
-                for dbmessage in dbmessages:
-                    previous_dialogue += dbmessage[0]+' '+dbmessage[1]+' '
-                # Update the string and preprocess it
-                quote = previous_dialogue + quote
-                quote = remove_non_english_russian_chars(quote)
-                quote = remove_extra_spaces(quote)
-                # Truncate the quote to return only the last MaxWords of words:
-                quote = " ".join(quote.split()[-config.MaxWords:])
-
-            # Fix the quote string, to eliminate errors:
-            quote = fixString(quote)
-
-            FailureCounter = 0 # In case there is a bug in the model
-            while (not message) and (FailureCounter<3):
-                message = WarClient.getReply(message=quote)
-                # Strange error in message if there is '02' in the message text.
-                if '02' in message:
-                    message = ""
-                FailureCounter+=1
-
-            if FailureCounter == 3:
-                continue # Skip that answer
-
-            # Post-processing fixes:
-            message = fixString(message)
-            print('Reply: ', message)
-
-            if message.endswith('.png'): # It is an image reply:
-                # Post an image reply:
-                login(username=config.username, password=config.password, thread_url=config.thread_url)
-                time.sleep(1)
-                post(message="", thread_url=config.thread_url, post_url=config.post_url, quoted_by=msg['messengerName'],
-                     quote_text=originalQuote, quote_source=msg['messageID'],
-                     img_url=message)
-                # will not be added to the database, if image is a reply
-
-            else:
-
-                # Add the new conversation pair to the database
-                db.setmessages(username=msg['messengerName'], message_text=originalQuote, bot_reply=message)
-                # Clean up the excessive records, leaving only the remaining messages
-                db.cleanup(username=msg['messengerName'], remaining_messages=config.remaining_messages)
-                # Delete the duplicate records
-                db.deleteDuplicates()
-
-                login(username=config.username, password=config.password, thread_url=config.thread_url)
-                time.sleep(1)
-                post(message=message, thread_url=config.thread_url, post_url=config.post_url, quoted_by=msg['messengerName'], quote_text=originalQuote, quote_source=msg['messageID'])
+    try: # Try logging in
+        login(username=config.username, password=config.password, thread_url=config.thread_url)
 
-            time.sleep(10)  # Standby time for server load release
+        lookUpPages = 5  # How many pages back to look in the thread
+        startingPage = getLastPage(thread_url=config.thread_url) - lookUpPages
+        if startingPage < 1:
+            startingPage = 1  # Starting page cannot be less than 1
+
+        # All messages (with quotes) by ALL users:
+        allMessages = getMessages(thread_url=config.thread_url, quotedUser='', startingPage=startingPage)
+
+        # IDs of the quoted messages, replied by the bot:
+        messages_by_bot_IDs = []
+
+        # Initiate the direct messages
+        direct_messages = []
+
+        for msg in allMessages:
+            # Direct message to the bot
+            if msg['quotedID'].split(': ')[-1] == '0': #debug
+                direct_messages.append(msg)
+            # Set a list of replied messages IDs
+            if msg['messengerName'] == config.username: #message posted by the WarBot
+                messages_by_bot_IDs.append(msg['quotedID'].split(': ')[-1])
+        # remove empty and repeated elements
+        messages_by_bot_IDs = list(set([elem for elem in messages_by_bot_IDs if elem]))
+
+        # All messages (with quotes) sent _FOR_ the Bot:
+        messagesForBot = getMessages(thread_url=config.thread_url, quotedUser=config.username, startingPage=startingPage)
+
+        # Append the direct messages to the messagesForBot:
+        for msg in direct_messages:
+            messagesForBot.append(msg)
+
+        # IDs of the messages, quoting the bot:
+        messages_for_bot_IDs = []
+
+        for msg in messagesForBot:
+            # Set a list of posted message IDs
+            messages_for_bot_IDs.append(msg['messageID'].split('-')[-1])
+        # remove empty elements
+        messages_for_bot_IDs = [elem for elem in messages_for_bot_IDs if elem]
+
+        # Filter to leave just the unanswered messages IDs:
+        messages_for_bot_IDs = [ID for ID in messages_for_bot_IDs if ID not in messages_by_bot_IDs]
+
+
+        # Reply the unanswered messages:
+        for msg in messagesForBot:
+            if msg['messageID'].split('-')[-1] in messages_for_bot_IDs:
+
+                originalQuote = msg['reply']
+                if originalQuote == "": # Just images, no text
+                    continue
+                else:
+                    quote = remove_non_english_russian_chars(msg['reply'])
+                    quote = remove_extra_spaces(quote)
+
+                message = "" #Initiating the reply message by Bot
+                previous_dialogue = "" #Initiating the previous dialogue
+
+                print('Quote: ', originalQuote)
+
+                # Init Connection
+                db = conversationDB.DataBase()
+
+                if msg['quotedID'].split(': ')[-1] != '0':  # It is dialogue. Look-up for the previous quotes
+
+                    # Get the previous dialogue from the database
+                    dbmessages = db.getmessages(msg['messengerName'])
+                    for dbmessage in dbmessages:
+                        previous_dialogue += dbmessage[0]+' '+dbmessage[1]+' '
+                    # Update the string and preprocess it
+                    quote = previous_dialogue + quote
+                    quote = remove_non_english_russian_chars(quote)
+                    quote = remove_extra_spaces(quote)
+                    # Truncate the quote to return only the last MaxWords of words:
+                    quote = " ".join(quote.split()[-config.MaxWords:])
+
+                # Fix the quote string, to eliminate errors:
+                quote = fixString(quote)
+
+                FailureCounter = 0 # In case there is a bug in the model
+                while (not message) and (FailureCounter<3):
+                    message = WarClient.getReply(message=quote)
+                    # Strange error in message if there is '02' in the message text.
+                    if '02' in message:
+                        message = ""
+                    FailureCounter+=1
+
+                if FailureCounter == 3:
+                    continue # Skip that answer
+
+                # Post-processing fixes:
+                message = fixString(message)
+                print('Reply: ', message)
+
+                if message.endswith('.png'): # It is an image reply:
+                    # Post an image reply:
+                    login(username=config.username, password=config.password, thread_url=config.thread_url)
+                    time.sleep(1)
+                    post(message="", thread_url=config.thread_url, post_url=config.post_url, quoted_by=msg['messengerName'],
+                         quote_text=originalQuote, quote_source=msg['messageID'],
+                         img_url=message)
+                    # will not be added to the database, if image is a reply
+
+                else:
+
+                    # Add the new conversation pair to the database
+                    db.setmessages(username=msg['messengerName'], message_text=originalQuote, bot_reply=message)
+                    # Clean up the excessive records, leaving only the remaining messages
+                    db.cleanup(username=msg['messengerName'], remaining_messages=config.remaining_messages)
+                    # Delete the duplicate records
+                    db.deleteDuplicates()
+
+                    login(username=config.username, password=config.password, thread_url=config.thread_url)
+                    time.sleep(1)
+                    post(message=message, thread_url=config.thread_url, post_url=config.post_url, quoted_by=msg['messengerName'], quote_text=originalQuote, quote_source=msg['messageID'])
+
+                time.sleep(10)  # Standby time for server load release
+                return 0 # All is good
+    except:
+        print('Bad Connection')
+        return -1 # Error
 
 
 if __name__ == '__main__':
@@ -346,7 +355,7 @@ if __name__ == '__main__':
     # Start the scheduler
     while True:
         print('Starting Session')
-        WarOnlineBot()
+        result = WarOnlineBot()
 
         # Debug Only:
         #imgWord = 'как выглядит'
@@ -356,8 +365,12 @@ if __name__ == '__main__':
         post(message="", thread_url=config.thread_url, post_url=config.post_url, quoted_by='Test',
              quote_text="posting an image",img_url='https://replicate.delivery/pbxt/knKBiJt8DPZ0B1o25PaLJSZjgv3D5HcwLoBIn0JESbe3nISIA/out-0.png')
         """
-        print('Session finished. Timeout...')
+        if result == 0: # Good result
+            print('Session finished. Timeout...')
 
-        timer = range(60 * config.timeout)
-        for t in timer:
-            time.sleep(1)
+            timer = range(60 * config.timeout)
+            for t in timer:
+                time.sleep(1)
+        else:
+            # Rerunning Session
+            time.sleep(10) # Standby time for server load release

config.py

@@ -35,7 +35,7 @@ MaxWords = 50 # The server is relatively weak to fast-process the long messages
 remaining_messages = 2
 
 # Time between the reply sessions:
-timeout = 5 # min
+timeout = 3 # min
 
 # Call for image generation:
 imgWord = 'как выглядит'