import time
import common.http
import asyncio
from cryptography.hazmat.primitives.serialization import load_pem_private_key
from cryptography.hazmat.backends import openssl
from cryptography.hazmat.primitives.asymmetric.padding import PKCS1v15
from cryptography.hazmat.primitives.hashes import SHA256
import json
import base64
import xml.dom
import xml.dom.minidom
def base64_encode(data):
return base64.urlsafe_b64encode(data).strip(b"=")
async def get_oauth_token(scopes):
with open("keys.json") as f:
keys = json.load(f)
t = int(time.time())
header = json.dumps({"alg":"RS256", "typ":"JWT"}).encode("utf-8")
claim = json.dumps({
"iss": keys["client_email"],
"scope": " ".join(scopes),
"aud": "https://accounts.google.com/o/oauth2/token",
"iat": t,
"exp": t+60*60,
}).encode("utf-8")
data = base64_encode(header) + b'.' + base64_encode(claim)
key = load_pem_private_key(keys["private_key"].encode("utf-8"), None, openssl.backend)
signature = key.sign(data, PKCS1v15(), SHA256())
jwt = (data + b'.' + base64_encode(signature)).decode("utf-8")
data = {"grant_type": "urn:ietf:params:oauth:grant-type:jwt-bearer", "assertion": jwt}
ret = json.loads((await common.http.request_coro("https://oauth2.googleapis.com/token", data, "POST")))
if "error" in ret:
raise Exception(ret["error"])
return ret
def find_schema(root, schema):
for link in root.getElementsByTagName("link"):
if link.attributes["rel"].value == schema:
return link.attributes["href"].value
def new_field(doc, name, value):
name = "gsx:"+"".join(filter(str.isalnum, name)).lower()
node = doc.createElement(name)
node.appendChild(doc.createTextNode(value))
return node
async def add_rows_to_spreadsheet(spreadsheet, rows):
token = await get_oauth_token(["https://spreadsheets.google.com/feeds"])
headers = {"Authorization": "%(token_type)s %(access_token)s" % token}
url = "https://spreadsheets.google.com/feeds/worksheets/%s/private/full" % spreadsheet
tree = xml.dom.minidom.parseString((await common.http.request_coro(url, headers=headers)))
worksheet = next(iter(tree.getElementsByTagName("entry")))
list_feed = find_schema(worksheet, "http://schemas.google.com/spreadsheets/2006#listfeed")
if list_feed is None:
raise Exception("List feed missing.")
list_feed = xml.dom.minidom.parseString((await common.http.request_coro(list_feed, headers=headers)))
post_url = find_schema(list_feed, "http://schemas.google.com/g/2005#post")
if post_url is None:
raise Exception("POST URL missing.")
for row in rows:
doc = xml.dom.minidom.getDOMImplementation().createDocument(None, "entry", None)
root = doc.documentElement
root.setAttribute("xmlns", "http://www.w3.org/2005/Atom")
root.setAttribute("xmlns:gsx", "http://schemas.google.com/spreadsheets/2006/extended")
for column, value in row:
root.appendChild(new_field(doc, column, value))
headers["Content-Type"] = "application/atom+xml"
await common.http.request_coro(post_url, headers=headers, data=doc.toxml(), method="POST")
