Add global variables for received and sent messages, print them before exiting
This commit is contained in:
parent
b037748b5e
commit
89472538eb
|
@ -8,6 +8,8 @@ import multiprocessing, urllib.request, urllib.error, urllib.parse, time, os
|
||||||
import dbtools, phototools, setuptools
|
import dbtools, phototools, setuptools
|
||||||
|
|
||||||
currentRun = []
|
currentRun = []
|
||||||
|
rec = 0
|
||||||
|
snt = 0
|
||||||
|
|
||||||
def status(driver):
|
def status(driver):
|
||||||
if "/main/login.php" not in driver.page_source:
|
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]
|
currentRun += [mid]
|
||||||
|
|
||||||
def pageHandler(driver, db = dbtools.dbHelper()):
|
def pageHandler(driver, db = dbtools.dbHelper()):
|
||||||
global currentRun
|
global currentRun, rec, snt
|
||||||
count = 0
|
count = 0
|
||||||
juha = BeautifulSoup(driver.page_source, "html5lib")
|
juha = BeautifulSoup(driver.page_source, "html5lib")
|
||||||
|
|
||||||
|
@ -88,8 +90,10 @@ def pageHandler(driver, db = dbtools.dbHelper()):
|
||||||
if not db.checkID(mid):
|
if not db.checkID(mid):
|
||||||
if mode == 0:
|
if mode == 0:
|
||||||
messageHandler(user, setuptools.user(), mid, date, driver, mode, db)
|
messageHandler(user, setuptools.user(), mid, date, driver, mode, db)
|
||||||
|
rec += 1
|
||||||
else:
|
else:
|
||||||
messageHandler(setuptools.user(), user, mid, date, driver, mode, db)
|
messageHandler(setuptools.user(), user, mid, date, driver, mode, db)
|
||||||
|
snt += 1
|
||||||
count += 1
|
count += 1
|
||||||
except IndexError:
|
except IndexError:
|
||||||
pass
|
pass
|
||||||
|
@ -124,3 +128,5 @@ if __name__ == "__main__":
|
||||||
p.terminate()
|
p.terminate()
|
||||||
p.join()
|
p.join()
|
||||||
driver.close()
|
driver.close()
|
||||||
|
print("Received", rec, "messages")
|
||||||
|
print("Sent", snt, "messages")
|
||||||
|
|
Loading…
Reference in a new issue