FilesExpand file tree

main

/

workflows.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

61 lines (48 loc) · 2.11 KB

main

/

workflows.py

Copy path

Top

File metadata and controls

• Code
• Blame

61 lines (48 loc) · 2.11 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

from server import PromptServer

from aiohttp import web

import os

import inspect

import json

import importlib

import sys

sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))

import pysssss

root_directory = os.path.dirname(inspect.getfile(PromptServer))

workflows_directory = os.path.join(root_directory, "pysssss-workflows")

workflows_directory = pysssss.get_config_value(

"workflows.directory", workflows_directory)

if not os.path.isabs(workflows_directory):

workflows_directory = os.path.abspath(os.path.join(root_directory, workflows_directory))

NODE_CLASS_MAPPINGS = {}

NODE_DISPLAY_NAME_MAPPINGS = {}

@PromptServer.instance.routes.get("/pysssss/workflows")

async def get_workflows(request):

files = []

for dirpath, directories, file in os.walk(workflows_directory):

for file in file:

if (file.endswith(".json")):

files.append(os.path.relpath(os.path.join(

dirpath, file), workflows_directory))

return web.json_response(list(map(lambda f: os.path.splitext(f)[0].replace("\\", "/"), files)))

@PromptServer.instance.routes.get("/pysssss/workflows/{name:.+}")

async def get_workflow(request):

file = os.path.abspath(os.path.join(

workflows_directory, request.match_info["name"] + ".json"))

if os.path.commonpath([file, workflows_directory]) != workflows_directory:

return web.Response(status=403)

return web.FileResponse(file)

@PromptServer.instance.routes.post("/pysssss/workflows")

async def save_workflow(request):

json_data = await request.json()

file = os.path.abspath(os.path.join(

workflows_directory, json_data["name"] + ".json"))

if os.path.commonpath([file, workflows_directory]) != workflows_directory:

return web.Response(status=403)

if os.path.exists(file) and ("overwrite" not in json_data or json_data["overwrite"] == False):

return web.Response(status=409)

sub_path = os.path.dirname(file)

if not os.path.exists(sub_path):

os.makedirs(sub_path)

with open(file, "w") as f:

f.write(json.dumps(json_data["workflow"]))

return web.Response(status=201)