0x2620/pandora_t_for_time
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

more advanced fragment parsing

Browse source
This commit is contained in:
j 2023-11-08 11:01:20 +01:00
parent 3252392eb5
commit 841a654254
1 changed files with 28 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
render.py
View file

@ -282,7 +282,7 @@ def render(root, scene, prefix=''):
files.append(path) files.append(path)
return files return files
def get_fragments(clips, voice_over): def get_fragments(clips, voice_over, prefix):
import itemlist.models import itemlist.models
import item.models import item.models
from collections import defaultdict from collections import defaultdict
@ -293,14 +293,37 @@ def get_fragments(clips, voice_over):
if l.name.split(' ')[0].isdigit(): if l.name.split(' ')[0].isdigit():
fragment = { fragment = {
'name': l.name, 'name': l.name,
'tags': [t['value'] for t in l.query['conditions'][1]['conditions'] if t['operator'] == '=='], 'tags': [],
'anti-tags': [t['value'] for t in l.query['conditions'][1]['conditions'] if t['operator'] == '!='], 'anti-tags': [],
'description': l.description 'description': l.description
} }
for con in l.query['conditions']:
if "conditions" in con:
for sub in con["conditions"]:
if sub['key'] == "tags" and sub['operator'] == '==':
fragment['tags'].append(sub['value'])
elif sub['key'] == "tags" and sub['operator'] == '!=':
fragment['tags'].append(sub['value'])
else:
print('unknown sub condition', sub)
elif con.get('key') == "tags" and con['operator'] == '==':
fragment['tags'].append(con['value'])
elif con.get('key') == "tags" and con['operator'] == '!=':
fragment['anti-tags'].append(con['value'])
fragment["id"] = int(fragment['name'].split(' ')[0]) fragment["id"] = int(fragment['name'].split(' ')[0])
originals = []
for i in l.get_items(l.user):
orig = i.files.filter(selected=True).first()
if orig:
ext = os.path.splitext(orig.data.path)[1]
type_ = i.data['type'][0].lower()
target = os.path.join(prefix, type_, i.data['title'] + ext)
originals.append(target)
fragment['clips'] = [] fragment['clips'] = []
for clip in clips: for clip in clips:
if set(clip['tags']) & set(fragment['tags']) and not set(clip['tags']) & set(fragment['anti-tags']): #if set(clip['tags']) & set(fragment['tags']) and not set(clip['tags']) & set(fragment['anti-tags']):
if clip['original'] in originals:
fragment['clips'].append(clip) fragment['clips'].append(clip)
fragment["voice_over"] = voice_over.get(str(fragment["id"]), {}) fragment["voice_over"] = voice_over.get(str(fragment["id"]), {})
fragments.append(fragment) fragments.append(fragment)
@ -322,7 +345,7 @@ def render_all(options):
clips = json.load(fd) clips = json.load(fd)
with open(os.path.join(prefix, "voice_over.json")) as fd: with open(os.path.join(prefix, "voice_over.json")) as fd:
voice_over = json.load(fd) voice_over = json.load(fd)
fragments = get_fragments(clips, voice_over) fragments = get_fragments(clips, voice_over, prefix)
with open(os.path.join(prefix, "fragments.json"), "w") as fd: with open(os.path.join(prefix, "fragments.json"), "w") as fd:
json.dump(fragments, fd, indent=2, ensure_ascii=False) json.dump(fragments, fd, indent=2, ensure_ascii=False)
position = target_position = 0 position = target_position = 0