Skip to content

Commit

Permalink
fix: several critical server fixes:
Browse files Browse the repository at this point in the history
* Resolved datetime conversion issues.
* Fixed handler method naming conventions.
* Removed unnecessary print statements.
  • Loading branch information
davidtorres committed Aug 16, 2024
1 parent cbfb778 commit 99ae62c
Showing 1 changed file with 6 additions and 10 deletions.
16 changes: 6 additions & 10 deletions kafka-auth-local-server/kafka_gcp_credentials_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,16 @@
import datetime
import http.server
import json

import google.auth
import google.auth.crypt
import google.auth.jwt
import google.auth.transport.urllib3
import urllib3


_credentials, _project = google.auth.default()
_credentials, _project = google.auth.default(
scopes=['https://www.googleapis.com/auth/cloud-platform']
)
_http_client = urllib3.PoolManager()


Expand All @@ -40,13 +41,11 @@ def valid_credentials():


def get_jwt(creds):
print('Creds ' + creds.service_account_email)
return json.dumps(
dict(
exp=creds.expiry.timestamp(),
iss='Google',
iat=datetime.datetime.now(datetime.timezone.utc).timestamp(),
scope='kafka',
sub=creds.service_account_email,
)
)
Expand All @@ -61,8 +60,6 @@ def b64_encode(source):


def get_kafka_access_token(creds):
print('token: ' + creds.token)
print('payload ' + b64_encode(get_jwt(creds)))
return '.'.join(
[b64_encode(_HEADER), b64_encode(get_jwt(creds)), b64_encode(creds.token)]
)
Expand All @@ -71,7 +68,7 @@ def get_kafka_access_token(creds):
def build_message():
creds = valid_credentials()
expiry_seconds = (
creds.expiry - datetime.now(datetime.timezone.utc)
creds.expiry - datetime.datetime.now()
).total_seconds()
return json.dumps(
dict(
Expand All @@ -90,14 +87,13 @@ def _handle(self):
self.send_header('Content-type', 'application/json')
self.end_headers()
message = build_message().encode('utf-8')
print(message)
self.wfile.write(message)

def do_get(self):
def do_GET(self):
"""Handles GET requests."""
self._handle()

def do_post(self):
def do_POST(self):
"""Handles POST requests."""
self._handle()

Expand Down

0 comments on commit 99ae62c

Please sign in to comment.