Skip to content

Commit 8d7cb9b

Browse files
committed
python: Update examples for new API
1 parent c79e944 commit 8d7cb9b

File tree

4 files changed

+17
-37
lines changed

4 files changed

+17
-37
lines changed

python/examples/VaasExample/authentication_example.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,8 @@ async def main():
3232
token_endpoint=token_url
3333
)
3434

35-
async with Vaas(url=vaas_url) as vaas:
36-
await vaas.connect(await authenticator.get_token())
37-
url = "https://secure.eicar.org/eicar.com"
38-
verdict = await vaas.for_url(url)
39-
print(f"Url {url} is detected as {verdict['Verdict']}")
35+
# Use the authenticator in VaaS
36+
# Vaas(authenticator=authenticator)
4037

4138

4239
if __name__ == "__main__":

python/examples/VaasExample/main.py

Lines changed: 7 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,39 +1,30 @@
11
from argparse import _AttributeHolder
22
import asyncio
33
import os
4+
import dotenv
45
from vaas import Vaas, ClientCredentialsGrantAuthenticator
56

67

78
async def main():
9+
dotenv.load_dotenv()
810
token_url = os.getenv("TOKEN_URL")
911
vaas_url = os.getenv("VAAS_URL")
1012

1113
if token_url is None:
1214
token_url = "https://account.gdata.de/realms/vaas-production/protocol/openid-connect/token"
1315
if vaas_url is None:
14-
vaas_url = "wss://gateway.production.vaas.gdatasecurity.de"
16+
vaas_url = "https://gateway.production.vaas.gdatasecurity.de"
1517

1618
authenticator = ClientCredentialsGrantAuthenticator(
1719
os.getenv("CLIENT_ID"),
1820
os.getenv("CLIENT_SECRET"),
1921
token_endpoint=token_url
2022
)
21-
async with Vaas(url=vaas_url) as vaas:
22-
token = await authenticator.get_token()
23-
await vaas.connect(token)
24-
path = os.getenv("SCAN_PATH")
25-
verdict = await vaas.for_file(path)
26-
print(f"{verdict['Sha256']} is detected as {verdict['Verdict']}")
2723

28-
# The scan functions will return the following dict:
29-
# {
30-
# "Sha256": "<Sha256>",
31-
# "Guid": "<Guid>",
32-
# "Verdict": <"Clean"|"Malicious"|"Unknown"|"Pup">,
33-
# "Detection": "<Name of the detected malware if found>",
34-
# "FileType": "<FileType>",
35-
# "MimeType": "<MimeType>"
36-
# }
24+
vaas = Vaas(url=vaas_url, authenticator=authenticator)
25+
path = os.getenv("SCAN_PATH")
26+
verdict = await vaas.for_file(path)
27+
print(f"{verdict.sha256} is detected as {verdict.verdict}")
3728

3829

3930
if __name__ == "__main__":

python/examples/VaasExample/main_url.py

Lines changed: 7 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,37 +1,29 @@
11
import asyncio
22
import os
33
from vaas import Vaas, ClientCredentialsGrantAuthenticator
4+
import dotenv
45

56

67
async def main():
8+
dotenv.load_dotenv()
79
token_url = os.getenv("TOKEN_URL")
810
vaas_url = os.getenv("VAAS_URL")
911

1012
if token_url is None:
1113
token_url = "https://account.gdata.de/realms/vaas-production/protocol/openid-connect/token"
1214
if vaas_url is None:
13-
vaas_url = "wss://gateway.production.vaas.gdatasecurity.de"
15+
vaas_url = "https://gateway.production.vaas.gdatasecurity.de"
1416

1517
authenticator = ClientCredentialsGrantAuthenticator(
1618
os.getenv("CLIENT_ID"),
1719
os.getenv("CLIENT_SECRET"),
1820
token_endpoint=token_url
1921
)
20-
async with Vaas(url=vaas_url) as vaas:
21-
await vaas.connect(await authenticator.get_token())
22-
url = "https://secure.eicar.org/eicar.com"
23-
verdict = await vaas.for_url(url)
24-
print(f"Url {url} is detected as {verdict['Verdict']}")
2522

26-
# The scan functions will return the following dict:
27-
# {
28-
# "Sha256": "<Sha256>",
29-
# "Guid": "<Guid>",
30-
# "Verdict": <"Clean"|"Malicious"|"Unknown"|"Pup">,
31-
# "Detection": "<Name of the detected malware if found>",
32-
# "FileType": "<FileType>",
33-
# "MimeType": "<MimeType>"
34-
# }
23+
vaas = Vaas(url=vaas_url, authenticator=authenticator)
24+
url = "https://secure.eicar.org/eicar.com"
25+
verdict = await vaas.for_url(url)
26+
print(f"Url {url} is detected as {verdict.verdict}")
3527

3628

3729
if __name__ == "__main__":
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
gdata-vaas==5.2.0
1+
gdata-vaas==6.0.0

0 commit comments

Comments
 (0)