OBS-Skripte/AutoMoveReplay.py at master · Bootscreen/OBS-Skripte · GitHub

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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
#!/usr/bin/env python
# title             : AutoMoveReplay.py
# description       : Moves Replays to another Folder when the Replay Buffer is stopped
# author            : Bootscreen
# date              : 2019 05 11
# version           : 0.2
# dependencies      : - Python 3.6 (https://www.python.org/)
#                   :   - shutil
# notes             : Follow this step for this script to work:
#                   : Python:
#                   :   1. Install python (v3.6 and 64 bits, this is important)
#                   : OBS:
#                   :   1. Go to Tools › Scripts
#                   :   2. Click the "Python Settings" tab
#                   :   3. Select your python install path
#                   :   4. Click the "Scripts" tab
#                   :   5. Click the "+" button and add this script
#                   :   6. Set the Destinationpath
#                   :   7. Check "Enable"
#                   :
# python_version    : 3.6+
# ==============================================================================

import obspython as obs
import os, shutil

enabled = True
debug_mode = False
location_to = ""
location_from = ""
file_type = ""
replay_status = False
check_frequency = 1000

def script_defaults(settings):
    global debug_mode
    if debug_mode: print("Calling defaults")

    global enabled
    global location_to

    obs.obs_data_set_default_bool(settings, "enabled", enabled)
    obs.obs_data_set_default_bool(settings, "debug_mode", debug_mode)
    obs.obs_data_set_default_string(settings, "location_to", "")

def script_description():
    return "<b>Moves the Replays to an other location when the Replaybuffer is stopped<br/></b>" + \
        "<hr>"

def script_properties():
    global debug_mode
    if debug_mode: print("[AMR] Loaded properties.")

    props = obs.obs_properties_create()
    obs.obs_properties_add_bool(props, "enabled", "Enabled")
    obs.obs_properties_add_bool(props, "debug_mode", "Debug Mode")
    obs.obs_properties_add_path(props, "location_to", "Path where the Replays should be moved", obs.OBS_PATH_DIRECTORY, "", None )

    return props

def script_save(settings):
    global debug_mode
    if debug_mode: print("[AMR] Saved properties.")

    script_update(settings)

def script_load(settings):
    global replay_status
    global debug_mode
    if debug_mode: print("[AMR] Loaded script.")

    replay_status = obs.obs_frontend_replay_buffer_active()

def script_unload():
    global debug_mode
    if debug_mode: print("[AMR] Unloaded script.")

def script_update(settings):
    global debug_mode
    if debug_mode: print("[AMR] Updated properties.")

    global enabled
    global location_to
    global replay_status


    debug_mode = obs.obs_data_get_bool(settings, "debug_mode")
    location_to = obs.obs_data_get_string(settings, "location_to")

    if obs.obs_data_get_bool(settings, "enabled") is True:
        if (not enabled):
            if debug_mode: print("[AMR] Enabled replay status check timer.")

        enabled = True
        obs.timer_remove(check_replay_status)
        obs.timer_add(check_replay_status, check_frequency)
    else:
        if (enabled):
            if debug_mode: print("[AMR] Disabled replay status check timer.")

        enabled = False
        obs.timer_remove(check_replay_status)

    replay_status = obs.obs_frontend_replay_buffer_active()

def get_last_replay():
    replay_buffer = obs.obs_frontend_get_replay_buffer_output()
    cd = obs.calldata_create()
    ph = obs.obs_output_get_proc_handler(replay_buffer)
    obs.proc_handler_call(ph, "get_last_replay", cd)
    path = obs.calldata_string(cd, "path")
    obs.calldata_destroy(cd)

    obs.obs_output_release(replay_buffer)
    return path

def check_replay_status():
    if not enabled:
        return

    global replay_status
    global location_from
    global file_type

    global debug_mode
    if debug_mode: print("[AMR] check_replay_status.")

    current_status = obs.obs_frontend_replay_buffer_active()
    if current_status is False and replay_status is True:
        if debug_mode: print("[AMR] current_status is False and replay_status is True.")
        if len(location_from) <= 0 or len(file_type) <= 0:
            last_replay = get_last_replay()
            if last_replay is not None and len(last_replay) > 0:
                location_from = os.path.dirname(os.path.abspath(last_replay))
                filename, file_type = os.path.splitext(last_replay)

        if len(location_from) > 0 and os.path.exists(location_from) and len(location_to) > 0 and os.path.exists(location_to) :
            if debug_mode: print("[AMR] move replays.")
            sourcefiles = os.listdir(location_from)
            for file in sourcefiles:
                if file.endswith(file_type):
                    if debug_mode: print("[AMR] move ." + file)
                    try:
                        shutil.move(os.path.join(location_from,file), os.path.join(location_to,file))
                    except:
                        print("[AMR] error while moving: " + file)
    replay_status = current_status