From 129249995e9b9a51356c40573b6673a7a2495d92 Mon Sep 17 00:00:00 2001
From: Vladislav Yarmak <vladislav@vm-0.com>
Date: Sun, 9 Jun 2019 21:45:31 +0300
Subject: [PATCH] ndl: add WS{2016,2019}

---
 tools/nv-driver-locator/README.md             | 25 ++++++++++++++++++-
 .../nv-driver-locator/get_nvidia_downloads.py |  2 ++
 .../nv-driver-locator.json.sample             |  9 +++++++
 3 files changed, 35 insertions(+), 1 deletion(-)

diff --git a/tools/nv-driver-locator/README.md b/tools/nv-driver-locator/README.md
index ffd0f0f..f71de80 100644
--- a/tools/nv-driver-locator/README.md
+++ b/tools/nv-driver-locator/README.md
@@ -187,6 +187,29 @@ All scripts may be used both as standalone application and importable module. Fo
                 "product": "QuadroMobile",
                 "certlevel": "All"
             }
+        },
+        {
+            "type": "nvidia_downloads",
+            "name": "downloads win server quadro certified",
+            "params": {
+                "os": "WindowsServer2012R2_64",
+                "product": "Quadro",
+                "certlevel": "Certified"
+            }
+        },
+        {
+            "type": "nvidia_downloads",
+            "name": "downloads win server 2019 quadro certified",
+            "params": {
+                "os": "WindowsServer2019",
+                "product": "Quadro",
+                "certlevel": "Certified"
+            }
+        },
+        {
+            "type": "cuda_downloads",
+            "name": "cuda toolkit tracker",
+            "params": {}
         }
     ],
     "notifiers": [
@@ -261,7 +284,7 @@ Type: `nvidia_downloads`
 
 Params:
 
-* `os` - OS family, version and bitness. Allowed values: `Linux_32`, `Linux_64`, `Windows7_32`, `Windows7_64`, `Windows10_32`, `Windows10_64`, `WindowsServer2012R2_32`, `WindowsServer2012R2_64`. Default: `Linux_64`.
+* `os` - OS family, version and bitness. Allowed values: `Linux_32`, `Linux_64`, `Windows7_32`, `Windows7_64`, `Windows10_32`, `Windows10_64`, `WindowsServer2012R2_32`, `WindowsServer2012R2_64`, `WindowsServer2016`, `WindowsServer2019`. Default: `Linux_64`.
 * `product` - product kind. Allowed values: `GeForce`, `GeForceMobile`, `Quadro`, `QuadroMobile`. Default: `GeForce`.
 * `certlevel` - driver certification level. Allowed values: `All` - any certification level, `Beta` - beta drivers, `Certified` - WHQL certified in Windows case and Nvidia certified in Linux case, `ODE` - Optimal Driver for Enterprise (Quadro driver), `QNF` - Quadro New Feature (Quadro driver). Default: `All`.
 * `driver_type` - driver type. Allowed values: `Standard`, `DCH`. At this moment DCH driver appears to exists only for some product families and only for Windows 10 x64. Default: `Standard`.
diff --git a/tools/nv-driver-locator/get_nvidia_downloads.py b/tools/nv-driver-locator/get_nvidia_downloads.py
index b1c29a3..9df495d 100755
--- a/tools/nv-driver-locator/get_nvidia_downloads.py
+++ b/tools/nv-driver-locator/get_nvidia_downloads.py
@@ -21,6 +21,8 @@ class OS(enum.Enum):
     Windows10_64 = 57
     WindowsServer2012R2_32 = 32
     WindowsServer2012R2_64 = 44
+    WindowsServer2016 = 74
+    WindowsServer2019 = 119
 
     def __str__(self):
         return self.name
diff --git a/tools/nv-driver-locator/nv-driver-locator.json.sample b/tools/nv-driver-locator/nv-driver-locator.json.sample
index 8b53eed..39326c3 100644
--- a/tools/nv-driver-locator/nv-driver-locator.json.sample
+++ b/tools/nv-driver-locator/nv-driver-locator.json.sample
@@ -176,6 +176,15 @@
                 "certlevel": "Certified"
             }
         },
+        {
+            "type": "nvidia_downloads",
+            "name": "downloads win server 2019 quadro certified",
+            "params": {
+                "os": "WindowsServer2019",
+                "product": "Quadro",
+                "certlevel": "Certified"
+            }
+        },
         {
             "type": "cuda_downloads",
             "name": "cuda toolkit tracker",