###########################
#!/usr/bin/python

from __future__ import print_function

import requests
import re
import hashlib
import base64


def login(baseurl, username, password):
   s = requests.Session()
   r = s.get(baseurl + "html/index.html")
   csrf_tokens = grep_csrf(r.text)

   s.headers.update({
      '__RequestVerificationToken': csrf_tokens[0]
      })

      # test token on statistics api
      # r = s.get(baseurl + "api/monitoring/statistic-server")

   data = login_data(username, password, csrf_tokens[0])
   r = s.post(baseurl + "api/user/login", data=data)

   s.headers.update({
      '__RequestVerificationToken': r.headers["__RequestVerificationTokenone"]
      })

   return s


def reboot(baseurl, session):
   s.post(baseurl + "api/device/control", data='1')


def grep_csrf(html):
   pat = re.compile(r".*meta name=\"csrf_token\" content=\"(.*)\"", re.I)
   matches = (pat.match(line) for line in html.splitlines())
   return [m.group(1) for m in matches if m]


def login_data(username, password, csrf_token):
      def encrypt(text):
         m = hashlib.sha256()
         m.update(text)
      return base64.b64encode(m.hexdigest())

      password_hash = encrypt(username + encrypt(password) + csrf_token)

      return '%s%s4' % (username, password_hash)


WEB = "http://192.168.1.1/"
USERNAME = "admin"
PASSWORD = "admin"

#if __name__ == "__main__":
#   s = login(WEB, USERNAME, PASSWORD)
#reboot(WEB, s)
#########################