0x2620/dd-re
2
0
Fork 0
Code Issues 4 Pull requests Releases Wiki Activity

patch clip out points (next in point or end of scene)

Browse source
This commit is contained in:
rlx 2018-01-26 11:48:40 +01:00
parent 3d95161942
commit f1d55b89d4
1 changed files with 22 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
recommendation_engine.py
View file

@ -28,6 +28,26 @@ class Engine:
else:
self.playlists = []
def _patch_clips(self, clips):
inpoints = {}
for index, clip in enumerate(clips):
video_id = clip['id'].split('/')[0]
inpoints[video_id] = inpoints.get(video_id, []) + [{
'index': index,
'position': clip['in']
}]
for video_id in inpoints:
for i, inpoint in sorted(
inpoints[video_id], key=lambda inpoint: inpoint['position']
):
if i < len(inpoints[video_id]) - 1:
clips[inpoint['index']]['out'] = inpoints[video_id][i + 1]
else:
clips[inpoint['index']]['out'] self.pandora.get({
'id': id, keys: ['duration']
})['duration']
return clips
def _shift_clips(self, clips):
index = random.randrange(len(clips))
return clips[index:] + clips[:index - 1]
@ -119,11 +139,11 @@ class Engine:
'id': storyline['id'],
'name': storyline['nodename'],
'tags': storyline['tags'],
'clips': [{
'clips': self._patch_clips([{
'id': clip['id'],
'in': clip['in'],
'out': clip['out']
} for clip in clips if clip['value'] == storyline['name']]
} for clip in clips if clip['value'] == storyline['name']])
} for storyline in storylines] if playlist['clips']]
with open(os.path.join(self.path, 'playlists.json'), 'w') as f:
f.write(json.dumps(self.playlists, indent=4, sort_keys=True))