summaryrefslogtreecommitdiff
path: root/domestic/session/server_handling/image.py
blob: 15719e0158c34d9059095f7b7b881c151b0295f9 (plain)
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
from domestic.parse.error_exception_handling import *
from domestic.utility.validate_dict_key import *
from domestic.session.session_message import *


@error_exception_handling
def image(message):
  monitor = validate_dict_key(message, 'monitor')
  screenshot = validate_dict_key(message, 'screenshot')
  cam = validate_dict_key(message, 'cam')
  
  assert screenshot or cam

  if monitor is None:
    message['monitor'] = 0
  else:
    message['monitor'] = int(monitor)

  if screenshot:
    message['image_type'] = True
    del message['screenshot']
    image_type = 'screenshot'
  else:
    message['image_type'] = False
    del message['cam']
    image_type = 'cam-screenshot'

  data = session_message(message, False)

  if data['screenshot']:
    make_image(['image', f'image/{image_type}'], data['screenshot'], success_message=data['message'], image_type=message['image_type'])