Simple authentication script for Earth Engine API, including a test.

utils-test.py

import unittest

from eesetup import utils

class Test(unittest.TestCase):
    def setUp(self):
        pass
    
    def test_geeAuth(self):
        result = utils.geeAuth()
        expected = True
        
        self.assertEqual(result, expected)

utils.py

import ee

def geeAuth(key_path=None):
    """Authenticate Earth Engine user.  Ensure that the *.p12 key file
    is in your ~/.ssh/ directory.
    """
    if not key_path:
        key_path = os.path.expanduser(os.environ['GOOGLE_PRIVATE_KEY_FILE'])
    if os.path.exists(key_path):
        acct = os.environ['GOOGLE_SERVICE_ACCOUNT']
        ee.Initialize(ee.ServiceAccountCredentials(acct, key_path))
        return True
    else:
        raise Exception('Ensure GEE key file is in ~/.ssh directory')