Add global variables for received and sent messages, print them before exiting

This commit is contained in:
Klaus-Uwe Mitterer 2016-08-07 23:14:58 +02:00
parent b037748b5e
commit 89472538eb

View file

@ -8,6 +8,8 @@ import multiprocessing, urllib.request, urllib.error, urllib.parse, time, os
import dbtools, phototools, setuptools
currentRun = []
rec = 0
snt = 0
def status(driver):
if "/main/login.php" not in driver.page_source:
@ -70,7 +72,7 @@ def messageHandler(sender, recipient, mid, date, driver, mode = 0, db = dbtools.
currentRun += [mid]
def pageHandler(driver, db = dbtools.dbHelper()):
global currentRun
global currentRun, rec, snt
count = 0
juha = BeautifulSoup(driver.page_source, "html5lib")
@ -88,8 +90,10 @@ def pageHandler(driver, db = dbtools.dbHelper()):
if not db.checkID(mid):
if mode == 0:
messageHandler(user, setuptools.user(), mid, date, driver, mode, db)
rec += 1
else:
messageHandler(setuptools.user(), user, mid, date, driver, mode, db)
snt += 1
count += 1
except IndexError:
pass
@ -124,3 +128,5 @@ if __name__ == "__main__":
p.terminate()
p.join()
driver.close()
print("Received", rec, "messages")
print("Sent", snt, "messages")