Add info fetching script

Author: David Pickart

.gitignore

@@ -1,4 +1,5 @@
 .DS_Store
 *.pyc
 *.db
-
+*.p
+*creds.txt

fetchInfo.py

@@ -0,0 +1,60 @@
+import pickle, requests, re, datetime
+
+printerPages = {}
+session = requests.session()
+endpoint = 'https://print.ads.carleton.edu:9192/app'
+
+loginInfo = {
+    'service': 'direct/1/Home/$Form$0',
+    'sp': 'S0',
+    'Form0': '$Hidden$0,$Hidden$1,inputUsername,inputPassword,$PropertySelection$0,$Submit$0',
+    '$Hidden$0': 'true',
+    '$Hidden$1': 'X',
+    '$PropertySelection$0': 'en',
+    '$Submit$0': 'Log in'
+  }
+
+def login(u, p):
+	
+	loginInfo['inputUsername'] = u
+	loginInfo['inputPassword'] = p
+	session.get(endpoint)
+	session.post(endpoint, loginInfo)
+
+def navigateToPage():
+	print "navigating to page"
+	session.get(endpoint + '?service=page/UserWebPrint')
+	session.get(endpoint + '?service=action/1/UserWebPrint/0/$ActionLink')
+
+def storePrinterInfo(attempt):
+	if attempt <= 3:
+		print "Storing info on page " + str(attempt)
+		url = endpoint + '?service=direct/1/UserWebPrintSelectPrinter/table.tablePages.linkPage&sp=AUserWebPrintSelectPrinter%2Ftable.tableView&sp=' + str(attempt)
+		response = session.get(url)
+		lines = response.text.split('<label')
+		regex = '.*value=\"[0-9]+\".*\n(.*).*'
+		for line in lines:
+			matches = re.findall(regex, line)
+			if matches:
+				printerPages[str(matches[0])] = attempt
+
+		storePrinterInfo(attempt + 1)
+
+
+def main():
+
+	printerPages["updated"] = str(datetime.datetime.utcnow())
+
+	with open("creds.txt", "r") as file:
+		u, p = file.read().splitlines()
+	if not u or not p:
+		print "Please include credentials"
+		return
+
+	login(u, p)
+	navigateToPage()
+	storePrinterInfo(1)
+	pickle.dump(printerPages, open("printerPages.p", "wb"))
+
+if __name__ == '__main__':
+	main()

jfp_server.py

@@ -5,6 +5,7 @@
 from flask import Flask, request, session, g, redirect, url_for, \
      abort, render_template, flash, jsonify
 from contextlib import closing
+import pickle
 
 # configuration
 DATABASE = '/var/www/web-print-server/web-print-server/db/data.db'
@@ -17,80 +18,6 @@
 app = Flask(__name__)
 app.config.from_object(__name__)
 
-printerPages = {
-  "print\\CASS101-X4600": 1,
-  "print\\CLST100-CC5051": 1,
-  "print\\CMC020-CC5051":  1,
-  "print\\CMC104-CC5051":  1,
-  "print\\CMC104-Gray-CC5051": 1,
-  "print\\CMC121H-CC5051": 1,
-  "print\\CMC201-X4600": 1,
-  "print\\CMC305-X4600": 1,
-  "print\\COWL108-CC5051": 1,
-  "print\\FACI318-CC5051": 1,
-  "print\\GHUE156-X5550":  1,
-  "print\\GOOD102-X3610":  1,
-  "print\\GOOD103-CC5051": 1,
-  "print\\GOOD103-X6350":  1,
-  "print\\HOPN104-CC5051": 1,
-  "print\\HUL007A-X3600":  1,
-  "print\\HUL010-X3600": 1,
-  "print\\HUL014-X6350": 1,
-  "print\\HUL100-CC5051":  1,
-  "print\\LAIR115-CC5051": 1,
-  "print\\LAIR118-CC5051": 1,
-  "print\\LAIR208-CC5051": 1,
-  "print\\LAIR208-X4600":  1,
-  "print\\LAIR300-X3610":  1,
-  "print\\LAST110-CC5051": 1,
-  "print\\LDC220-CC5051":  2,
-  "print\\LDC243-X5550": 2,
-  "print\\LEIG128-LJM602": 2,
-  "print\\LEIG217-X4510":  2,
-  "print\\LEIG218-CC5051": 2,
-  "print\\LEIG218-Gray-CC5051":  2,
-  "print\\LEIG231-X4600":  2,
-  "print\\LEIG326-CC5051": 2,
-  "print\\LEIG326-X4600":  2,
-  "print\\LEIG414-LJM602": 2,
-  "print\\LIBR-Public-X5550":  2,
-  "print\\LIBR400-CC5051": 2,
-  "print\\MUDD075-X4510":  2,
-  "print\\MUDD169-X4600":  2,
-  "print\\MUSI200-X4600":  2,
-  "print\\OLIN007-X6350":  2,
-  "print\\OLIN011-X3600":  2,
-  "print\\OLIN104-X4510":  2,
-  "print\\OLIN112-X4510":  2,
-  "print\\OLIN125-CC5051": 2,
-  "print\\OLIN125-Gray-CC5051":  2,
-  "print\\OLIN215-CC5051": 2,
-  "print\\OLIN301-X4600":  2,
-  "print\\OLIN311-CC5051": 2,
-  "print\\OLIN311-X6350":  2,
-  "print\\RSC105-CC5051":  3,
-  "print\\RSC235-CC5051":  3,
-  "print\\SAYL-Public-X5550":  3,
-  "print\\SAYL050-X6360":  3,
-  "print\\SAYL057-CC5051": 3,
-  "print\\SAYL109A-X4510": 3,
-  "print\\SAYL150-X4510":  3,
-  "print\\SCOV014-CC5051": 3,
-  "print\\SEVY014-CC5051": 3,
-  "print\\SEVY129-CC5051": 3,
-  "print\\STRG107-CC5051": 3,
-  "print\\TWCO100-CC5051": 3,
-  "print\\WCC003-X3610": 3,
-  "print\\WCC028-CC5051":  3,
-  "print\\WCC138-X6360": 3,
-  "print\\WCC146-X3610": 3,
-  "print\\WCC225-CC5051":  3,
-  "print\\WEST200-CC5051": 3,
-  "print\\WILL119-X4600":  3,
-  "print\\WILL310-CC5051": 3,
-  "print\\WILL409-X4600":  3
-}
-
 #database helper functions
 def connect_db():
     return sqlite3.connect(app.config['DATABASE'])
@@ -163,7 +90,7 @@ def get_status():
                 response["errors"] = message
     except:
         pass
-    response["printerPages"] = printerPages
+    response["printerPages"] = pickle.load(open("printerPages.p", "rb"))
     return(jsonify(**response))