File size: 2,513 Bytes
f5776d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
import inspect
import string
import random
import requests
import json
class FuncInspector:
def __init__(self):
self.calls = []
def inspect_inner(self, func, function_calls):
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
self.merge_result(result, function_calls)
return result
return wrapper
def inspect_func(self, func):
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
stack = inspect.stack()
function_calls = []
for i in range(len(stack)):
if stack[i][3] == "<module>":
break
if stack[i][3] != "wrapper":
function_calls.append(stack[i][3])
function_calls.reverse()
result = self.inspect_inner(result, function_calls)
return result
return wrapper
def parse(self, obj, delete_empty=False):
if isinstance(obj, list):
for elem in obj:
self.parse(elem, delete_empty)
if isinstance(obj, dict):
to_delete = []
for key in obj:
if delete_empty and not obj[key]:
to_delete.append(key)
elif key == "completions":
to_delete.append(key)
else:
self.parse(obj[key], delete_empty)
for key in to_delete:
obj.pop(key)
def merge_result(self, result, function_calls):
prev_list = self.calls
prev_call = {} if not prev_list else prev_list[-1]
for call in function_calls[:-1]:
if call not in prev_call:
prev_call = {call: []}
prev_list.append(prev_call)
prev_list = prev_call[call]
prev_call = {} if not prev_list else prev_list[-1]
example_obj = result[0]
self.parse(example_obj)
prev_list.append({ function_calls[-1]: example_obj })
def view_data(self):
chars = string.digits + string.ascii_lowercase
id = ''.join(random.choices(chars, k=8))
post_url = 'http://127.0.0.1:5000/log-item'
parsed_calls = self.calls.copy()
self.parse(parsed_calls, delete_empty=True)
data = {'id': id, 'content': parsed_calls}
response = requests.post(post_url, json=data)
if response.status_code == 201:
print('Data created successfully')
else:
print(f'Error sending data to server: {response.status_code}')
return
frontend_url = f"http://localhost:3000?id={id}"
print(f"View the data here, {frontend_url}")
def output_json(self, out_path):
f = open(out_path, "w")
json_object = json.dumps(self.calls, indent=2)
f.write(json_object)
|