def loginTest(username, password, outputlog = True, browser = "PhantomJS"):
orig_stdout = sys.stdout # re-route output
logfile = None
if outputlog:
# use current time in log file name
logfilename = datetime.strftime(datetime.now(), '%Y-%m-%d %H:%M:%S')
logfilename = logfilename.replace(':', '_') + ".log"
logfile = open('../tmp/' + logfilename, 'w+')
sys.stdout = logfile
# input error handle
if username == [] or password == [] or len(username) != len(password):
print "username array does not have the same length as password array..."
# close log file
if outputlog:
sys.stdout = orig_stdout
logfile.close()
return
driver = getDriver(browser)
begintime = time.time()
# loop through all username/password combinations
for idx in range(len(username)):
eachbegintime = time.time()
print "--------------------------------------------------------------"
print "#", idx+1, "ID:", username[idx]
# just in case network connection is broken
try:
driver.get(amexWebsite)
except:
print "website is not available..."
if outputlog:
sys.stdout = orig_stdout
logfile.close() # close log file
return
# fill and submit login form
try:
amexLogIn(driver, username[idx], password[idx])
except:
print "Something is wrong with login"
continue # end current loop
closeFeedback(driver) # just in case the feedback banner appears
clickOnLoadMore(driver) # scroll down and click load more
# store offer names and click on offers
offernames = collectOfferNames(driver)
print "Available offers are:", offernames
if not offernames == '':
clickOnOffers(driver)
time.sleep(1)
# logout
try:
amexLogOut(driver)
except:
pass # pass would be fine, since the launch of the url is the log-in page no matter what
time.sleep(1)
eachendtime = time.time()
print "Time used: %0.2f seconds" % (eachendtime - eachbegintime)
print "--------------------------------------------------------------"
endtime = time.time()
# print summary
print "--------------------------------------------------------------"
print "** Summary **"
print "Total time used: %0.2f seconds" % (endtime - begintime)
print "--------------------------------------------------------------"
# close log file
if outputlog:
sys.stdout = orig_stdout
logfile.close()
# close browser
driver.quit()
def getAddedOffers(username, password, outputlog = True):
# re-route output
orig_stdout = sys.stdout
logfile = None
if outputlog:
# use current time as log file
logfilename = datetime.strftime(datetime.now(), '%Y-%m-%d %H:%M:%S')
logfilename = "offers " + logfilename.replace(':', '_') + ".csv"
logfile = open('../tmp/' + logfilename, 'w+')
sys.stdout = logfile
# input error handle
if username == [] or password == [] or len(username) != len(password):
print "username array does not have the same length as password array..."
# close log file
if outputlog:
sys.stdout = orig_stdout
logfile.close()
return
# use phantom JS / Firefox
# driver = webdriver.PhantomJS()
# driver = webdriver.Firefox()
driver = webdriver.Chrome('./chromedriver')
driver.maximize_window()
majorOfferDescMap = dict()
majorOfferDateMap = dict()
majorDateOfferPair = set()
userOffersList = []
# loop through all username/password combinations
for idx in range(len(username)):
eachbegintime = time.time()
try:
driver.get(amexWebsite)
except:
print "website is not available..."
# close log file
if outputlog:
sys.stdout = orig_stdout
logfile.close()
return
# fill and submit login form
try:
amexLogIn(driver, username[idx], password[idx])
except:
print "username/password combination is incorrect..."
continue
time.sleep(2)
closeFeedback(driver)
clickOnAddedToCard(driver)
clickOnLoadMore(driver)
# main program
offers = driver.find_elements_by_class_name("ah-card-offer")
offerstext = [offer.text.encode('utf-8') for offer in offers]
offersplit = [text.split('\n') for text in offerstext]
offerDescMap = dict()
offerDateMap = dict()
dateOfferPair = set()
offersSet = set()
# discard expired offers
for sp in offersplit:
expiration = dt.strptime(sp[3], '%m/%d/%Y')
if expiration > dt.now():
offerDescMap[sp[0]] = sp[1]
offerDateMap[sp[0]] = sp[3]
dateOfferPair.add((expiration, sp[0]))
offersSet.add(sp[0])
majorOfferDescMap.update(offerDescMap)
majorOfferDateMap.update(offerDateMap)
majorDateOfferPair.update(dateOfferPair)
userOffersList.append(offersSet)
time.sleep(1)
# logout
try:
amexLogOut(driver)
except:
pass
time.sleep(1)
majorDateOfferList = list(majorDateOfferPair)
majorDateOfferList.sort(key=lambda tup:tup[0])
# write 1st line
for dateOffer in majorDateOfferList:
offer = dateOffer[1].replace(',', '.')
sys.stdout.write(','+offer)
sys.stdout.write('\n')
# write 2nd line
for dateOffer in majorDateOfferList:
desc = majorOfferDescMap[dateOffer[1]].replace(',', ' and')
sys.stdout.write(','+desc)
sys.stdout.write('\n')
# write 3rd line
for dateOffer in majorDateOfferList:
date = majorOfferDateMap[dateOffer[1]]
sys.stdout.write(','+date)
sys.stdout.write('\n')
# write the rest
for i in range(len(username)):
sys.stdout.write(username[i])
for dateOffer in majorDateOfferList:
if dateOffer[1] in userOffersList[i]:
sys.stdout.write(',+')
else:
sys.stdout.write(',')
sys.stdout.write('\n')
# close log file
if outputlog:
sys.stdout = orig_stdout
logfile.close()
# close browser
driver.quit()
def loginTest(username, password, outputlog=True):
orig_stdout = sys.stdout # re-route output
logfile = None
if outputlog:
# use current time in log file name
logfilename = datetime.strftime(datetime.now(), '%Y-%m-%d %H:%M:%S')
logfilename = logfilename.replace(':', '_') + ".log"
logfile = open('../tmp/' + logfilename, 'w+')
sys.stdout = logfile
# input error handle
if username == [] or password == [] or len(username) != len(password):
print "username array does not have the same length as password array..."
# close log file
if outputlog:
sys.stdout = orig_stdout
logfile.close()
return
# use phantom JS
driver = webdriver.PhantomJS()
# driver = webdriver.Firefox()
# driver = webdriver.Chrome('./chromedriver')
driver.maximize_window()
begintime = time.time()
# loop through all username/password combinations
for idx in range(len(username)):
eachbegintime = time.time()
print "--------------------------------------------------------------"
print "#", idx + 1, "ID:", username[idx]
# just in case network connection is broken
try:
driver.get(amexWebsite)
except:
print "website is not available..."
if outputlog:
sys.stdout = orig_stdout
logfile.close() # close log file
return
# fill and submit login form
try:
amexLogIn(driver, username[idx], password[idx])
except:
print "Something is wrong with login"
continue # end current loop
closeFeedback(driver) # just in case the feedback banner appears
clickOnLoadMore(driver) # scroll down and click load more
# store offer names and click on offers
tmpoffernames = driver.find_elements_by_class_name(
"ah-card-offer-name") + driver.find_elements_by_class_name(
"ah-offer-name")
tmpnames = [n.text.encode('utf-8') for n in tmpoffernames]
tmpnames = filter(None, tmpnames)
print "Available offers are:", ', '.join(tmpnames)
clickOnOffers(driver)
# logout
try:
amexLogOut(driver)
except:
pass # pass would be fine, since the launch of the url is the log-in page no matter what
time.sleep(1)
eachendtime = time.time()
print "Time used: %0.2f seconds" % (eachendtime - eachbegintime)
print "--------------------------------------------------------------"
endtime = time.time()
# print summary
print "--------------------------------------------------------------"
print "** Summary **"
print "Total time used: %0.2f seconds" % (endtime - begintime)
print "--------------------------------------------------------------"
# close log file
if outputlog:
sys.stdout = orig_stdout
logfile.close()
# close browser
driver.quit()