Automower Cloud Function

automower_function.py

import os
import sys

from flask import abort, escape
from google.cloud import firestore

os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = 'sa_key.json'


def get_automower():
    db = firestore.Client()
    key = db.collection(u'automower').document(u'automower')
    snapshot = key.get()
    return snapshot.to_dict()['message']

def set_automower(message):
    db = firestore.Client()
    key = db.collection(u'automower').document(u'automower')
    key.set({
        'message': message
    })
    return 'Completed'

def main(request):
    if request.method == 'GET':
        return get_automower()
    elif request.method == 'POST':
        request_args = request.args
        return set_automower(request_args['message'])
    else:
        return abort(405)