feature(env update):

2023-10-31 20:05:29 +08:00 · 2023-10-31 20:05:29 +08:00 · de5a42d632
parent 429218fc02
commit de5a42d632
2 changed files with 3 additions and 4 deletions
--- 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
--- 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(