From d66c6a2ede0beb5e6193a7fc32328c2e241f9cf1 Mon Sep 17 00:00:00 2001
From: clericpy <clericpy@gmail.com>
Date: Sat, 2 May 2020 17:33:06 +0800
Subject: [PATCH] add some libs into _GLOBALS_ARGS for udf parser

---
 watchdogs/settings.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/watchdogs/settings.py b/watchdogs/settings.py
index d903c0f..eddc66d 100644
--- a/watchdogs/settings.py
+++ b/watchdogs/settings.py
@@ -81,6 +81,9 @@ def setup_models():
 
 async def setup_uniparser():
     import re
+    import datetime
+    import math
+    import random
     from torequests.utils import (curlparse, escape, guess_interval,
                                   itertools_chain, json, parse_qs, parse_qsl,
                                   ptime, quote, quote_plus, slice_by_size,
@@ -93,7 +96,10 @@ async def setup_uniparser():
     import uniparser.fastapi_ui
     UDFParser._GLOBALS_ARGS.update({
         're': re,
+        'datetime': datetime,
         'curlparse': curlparse,
+        'math': math,
+        'random': random,
         'escape': escape,
         'guess_interval': guess_interval,
         'itertools_chain': itertools_chain,