#!/usr/bin/env python2.7
# This script gets usage from my ISP's website, checks to see if the current average for the month 
# is above their recommended amount per day, then disables the wifi on my router if it is.
# I made this for a household of people who don't understand the consequences of their actions
# (namely, that if you abuse our awesome internet, we all have to suffer with being shaped :( )
# Please, modify this for your own ISP/router and set it up in your own house to help with your daily usage.
# -blha303
# BSD license.
username = "USERNAME"
password = "PASSWORD"
routerpass = "routerpass"
 
import urllib2
from BeautifulSoup import BeautifulSoup
import urllib
import cookielib
import time

def getnumbers(inp): # For parsing "100,000MB" to an integer
  return int(inp[:-2].replace(",", ""))

url = "https://myaccount2.westnet.com.au/" # login page url
desc = "westnet usage alert script by blha303. https://gist.github.com/blha303/5788199 for source" # what the server sees as the client ID

logindata = urllib.urlencode({"hdnLoginType": "myaccount", "action": "login", "username": username, "password": password}) # login POST data

cj = cookielib.CookieJar() # Cookie jar, used to store login session
opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cj))
opener.addheaders = [('User-Agent', desc)] # set client ID to desc

login = opener.open(url, logindata) # log in, save session data to cookie jar
 
page  = opener.open(url) # reopen same url (westnet has both login and homepage on the same url)
soup  = BeautifulSoup(page.read()) # make soup from page
usage = soup.findAll('div',{'class':'usage_text'})[0].text # get usage string "XXX,XXXMB out of XXX,XXXMB"
count = getnumbers(usage.split(" ")[0]) # get count integer, first part of usage string above as integer
total = getnumbers(usage.split(" ")[-1]) # get total integer, last part of usage string above as integer
suggested = getnumbers(soup.findAll('a',{'href':'#suggested_anytime_tooltip'})[0].text) # get suggested integer
trend = getnumbers(soup.findAll('a',{'href':'#trend_anytime_tooltip'})[0].text) # get trend integer
#print "usage: " + usage
#print "count: " + str(count)
#print "total: " + str(total)
#print "suggested: " + str(suggested)
#print "trend: " + str(trend)

boblogin = "http://10.1.1.1/login.cgi" # router login url
bobdata = urllib.urlencode({"login_option": "0", "password": routerpass, "passwordtemp": ""}) # router login info as POST data
wdisurl = "http://10.1.1.1/wireless_id.wl?wlSsidIdx=0&wlEnbl=0" # router disable wireless url as GET data
wenurl = "http://10.1.1.1/wireless_id.wl?wlEnbl=1&wlSsidIdx=0" # router enable wireless url as GET data

ts = time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.gmtime()) # timestamp for status webpage

def logintoBob():
  opener.open(boblogin, bobdata) # log into router, save session to cookie jar

def enableWireless(f):
  logintoBob()
  opener.open(wenurl) # send wireless enable request using login info from cookiejar
  f.write("%s: Enable wireless<br>" % ts)

def disableWireless(f):
  logintoBob()
  opener.open(wdisurl) # send wireless disable request using login info from cookiejar
  f.write("%s: Disable wireless<br>" % ts)

if count > total:
  shaped = True
else:
  shaped = False

#print "shaped: " + str(shaped)

f = open("/var/www/westnet.html", "a") # Append to web-accessible file
if trend > suggested and not shaped:
  disableWireless(f)
elif shaped or trend <= suggested:
  enableWireless(f)
f.close()