Create block_inactive_licenses.py

Author: artemlos

block_inactive_licenses.py

@@ -0,0 +1,90 @@
+# -*- coding: utf-8 -*-
+"""
+Created on Tue Feb  7 13:05:08 2023
+
+@author: Cryptolens AB
+"""
+
+import datetime
+import argparse
+import time
+from licensing.models import *
+from licensing.methods import Key, Helpers, Message, Product, Customer, Data, AI
+
+
+def identify_inactive_keys(token, productId, window=30, block=False):
+    
+    token = token
+    product_id=productId
+    
+    all_keys = []
+    
+    time_window = 60*60*24*window # 1 month back in time
+    time_now = int(time.time())
+    
+    filtered_keys = []
+    
+    not_processed_keys = []
+    
+    ## only non blocked keys should be listed.
+    
+    iteration = 1
+    while True:
+        temp = Product.get_keys(token, product_id, iteration)[0]
+        print("Processed page {0}".format(str(iteration)))
+        if len(temp) > 0:
+            all_keys.extend(temp)
+            iteration+=1
+        else:
+            break
+        
+    for key in all_keys:
+        
+        req = []
+        try:
+            req = AI.get_web_api_log(token, product_id, key["key"], limit=1, order_by="Id descending")[0]
+        except:
+            not_processed_keys.append(key["key"])
+            continue
+        
+        if len(req) > 0:
+            req = req[0]
+        else:
+            print("Skipped key {0}".format(key["key"]))
+            continue
+        
+        if req["time"] + time_window > time_now:
+            filtered_keys.append(req["key"])
+            print("Detected key {0}".format(req["key"]))
+        else:
+            print("Skipped key {0}".format(req["key"]))
+            
+    if block:
+        
+        for key in filtered_keys:
+            try:
+                res = Key.block_key(token, product_id, key)
+                if res[0] ==True:
+                    print("Key {0} was blocked successfully.".format(key))
+                else:
+                    print("Key {0} was not blocked.".format(key))
+            except:
+                print("Key {0} was not blocked.".format(key))
+            
+
+if __name__ == "__main__":
+
+    parser = argparse.ArgumentParser(description='This script will help you to identify which keys are not active and, if desired, block them.')
+    
+    parser.add_argument('-t', '--token', help="The access token with GetWebAPILog, GetKeys and BlockKey permissions.", required=True)
+    parser.add_argument('-p','--product', help="The ProductId  of the product", required=True)
+    parser.add_argument('-b','--block', help="If set to true, all inactive keys will be blocked.",action="store_true")
+    parser.add_argument("-w", '--window', help="How many days should the key have been inactive to mark it as inactive and eventually block it. Default 30 days.", default=30)
+    
+    args = parser.parse_args()
+    
+    print(identify_inactive_keys(args.token, args.product,args.window, args.block))
+        
+
+    
+