From de5a42d632daf8b2ce4e4710dcecf7834407b676 Mon Sep 17 00:00:00 2001
From: Devoalda <linuxes_mojoworld@aleeas.com>
Date: Tue, 31 Oct 2023 20:05:29 +0800
Subject: [PATCH] feature(env update):

---
 safeshare/docker-compose.yml      | 2 +-
 safeshare/safeshare_vdb/server.py | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/safeshare/docker-compose.yml b/safeshare/docker-compose.yml
index 0413543..8924ae1 100644
--- a/safeshare/docker-compose.yml
+++ b/safeshare/docker-compose.yml
@@ -51,7 +51,7 @@ services:
     ports:
       - "50051:50051"
     environment:
-      - API_TOKEN=YOUR_VIRUSTOTAL_API_TOKEN
+      - VIRUSTOTAL_API_KEY=YOUR_VIRUSTOTAL_API_TOKEN
       - AWS_ACCESS_KEY_ID=YOUR_AWS_ACCESS_KEY_ID
       - AWS_SECRET_ACCESS_KEY=YOUR_AWS_SECRET_ACCESS_KEY
       - REGION=YOUR_AWS_REGION
diff --git a/safeshare/safeshare_vdb/server.py b/safeshare/safeshare_vdb/server.py
index 682ee22..5c816a9 100644
--- a/safeshare/safeshare_vdb/server.py
+++ b/safeshare/safeshare_vdb/server.py
@@ -10,9 +10,8 @@ import requests
 
 import boto3 as boto  # 1.28.68
 
-# TotalVirus API key
-environ.Env.read_env('./.env')
-api = environ.Env().str('API_TOKEN')
+# VirusTotal API key
+api = os.environ.get('VIRUSTOTAL_API_KEY')
 
 # dynamo db instance
 session = boto.Session(