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
|
#!/bin/env python3
# -*- coding: utf-8 -*-
import argparse
import logging
import sys
from obswebsocket import obsws, requests
def __call(ws, req, cb=None):
r = ws.call(req)
if r.status:
logging.debug('Success calling %s' % req)
if cb:
try:
return cb(r)
except Exception as e:
raise(e)
else:
raise Exception(r)
def listActions(*_):
for a in actions:
print(a)
def startRecording(ws, _):
__call(ws, requests.StartRecording())
def stopRecording(ws, _):
__call(ws, requests.StopRecording())
__call(ws, requests.GetRecordingFolder(),
lambda r: print('Saved recording at: %s' % r.getRecFolder()))
def startStreaming(ws, _):
__call(ws, requests.StartStreaming())
def stopStreaming(ws, _):
__call(ws, requests.StopStreaming())
def listScenes(ws, _):
def cb(r):
current_scene = r.getCurrentScene()
for scene in r.getScenes():
scene = scene['name']
print('%s %s' % (('*' if current_scene == scene else ' '), scene))
__call(ws, requests.GetSceneList(), cb)
def switchScene(ws, args):
def cb(r):
return list(map(lambda s: s['name'], r.getScenes()))
scenes = __call(ws, requests.GetSceneList(), cb)
if len(args) != 1:
raise Exception('need exactly 1 argument')
if args[0] in scenes:
__call(ws, requests.SetCurrentScene(args[0]))
sys.path.append('../')
actions = {
'list-actions': listActions,
'start-recording': startRecording,
'stop-recording': stopRecording,
'start-streaming': startStreaming,
'stop-streaming': stopStreaming,
'list-scenes': listScenes,
'switch-scene': switchScene,
}
if __name__ == '__main__':
try:
app = argparse.ArgumentParser(
description=__doc__,
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
app.add_argument('-d', '--debug',
action='store_true',
help='show debug logs')
app.add_argument('-H', '--host', type=str,
default='localhost',
help='obs-websocket host')
app.add_argument('-P', '--port', type=int,
default=4444,
help='obs-websocket port')
app.add_argument('-p', '--password', type=str,
help='obs-websocket password')
app.add_argument('action', type=str,
choices=actions.keys(),
help='the command to run')
app.add_argument('args',
nargs=argparse.REMAINDER,
help='optional action argument')
args = app.parse_args()
if args.debug:
logging.basicConfig(level=logging.DEBUG)
ws = obsws(args.host, args.port, args.password)
ws.connect()
actions[args.action](ws, args.args)
ws.disconnect()
exit(0)
except KeyboardInterrupt:
pass
except Exception as e:
logging.error(e)
|
