New template (#21)

* add get_input_device_by_id method and delete useless template

* fix avaliability problem

* fix method name

* fix test and lint

* fix preview exception raising logic

* add data file to package

* add new param to template

* remove 'sourceType' param in 'generate_preview' method

* update version

Author: Spider-Min

elemental/client.py

@@ -185,7 +185,7 @@ def get_input_device_by_id(self, input_device_id):
         device_info.pop('@href')
         return dict(device_info)
 
-    def generate_preview(self, source_type, input_id):
+    def generate_preview(self, input_id):
         url = f'{self.server_ip}/inputs/generate_preview'
         headers = self.generate_headers(url)
 
@@ -195,7 +195,7 @@ def generate_preview(self, source_type, input_id):
 
         # generate body
         data = f"input_key=0&live_event[inputs_attributes][0][source_type]=" \
-               f"{source_type}&live_event[inputs_attributes][0]" \
+               f"DeviceInput&live_event[inputs_attributes][0]" \
                f"[device_input_attributes][sdi_settings_attributes]" \
                f"[input_format]=Auto&live_event[inputs_attributes][0]" \
                f"[device_input_attributes][device_id]={input_id}"
@@ -204,7 +204,7 @@ def generate_preview(self, source_type, input_id):
 
         response_parse = ast.literal_eval(response.text)
 
-        if 'preview_image_id' not in response_parse:
+        if 'type' in response_parse and response_parse['type'] == 'error':
             raise ElementalException(
                 f"Response: {response.status_code}\n{response.text}")
         else:

elemental/client_test.py

@@ -141,7 +141,17 @@ def test_create_event_should_call_send_request_as_expect_and_return_event_id():
                                     'mediastore_container_backup':
                                     'https://hu5n3jjiyi2jev.data.medias'
                                     'tore.us-east-1.amazonaws.com/backup',
-                                    'channel': "1", 'device_name': "0"})
+                                    'input_device': {'id': '1',
+                                                     'name': None,
+                                                     'device_name': 'HD-SDI 1',
+                                                     'device_number': '0',
+                                                     'device_type': 'AJA',
+                                                     'description':
+                                                         'AJA Capture Card',
+                                                     'channel': '1',
+                                                     'channel_type': 'HD-SDI',
+                                                     'quad': 'false',
+                                                     'availability': False}})
 
     response_from_elemental_api = client.send_request.call_args_list[0][1]
     assert response_from_elemental_api['http_method'] == 'POST'
@@ -347,7 +357,7 @@ def test_get_preview_will_parse_response_json_as_expect():
         status=200, text=file_fixture(
             'success_response_for_generate_preview.json'))
 
-    response = client.generate_preview('DeviceInput', '2')
+    response = client.generate_preview('2')
 
     assert response == {
         'preview_url': f'{ELEMENTAL_ADDRESS}/'
@@ -367,7 +377,7 @@ def test_get_preview_will_raise_ElementalException_if_preview_unavaliable():
                                                 "Device already in use."}))
 
     with pytest.raises(ElementalException) as exc_info:
-        client.generate_preview('DeviceInput', '1')
+        client.generate_preview('1')
 
     respond_text = json.dumps({'type': 'error',
                                'message': 'Input is invalid. '

elemental/templates/qvbr_mediastore.xml

@@ -1,6 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <live_event href="/live_events/3" product="Elemental Live + HEVC Package" version="2.14.4.308149">
-  <name>SuperBowl AVC Live (FLL - Dev)</name>
+  <name>{{channel_name}}</name>
   <input>
     <deblock_selected>true</deblock_selected>
     <denoise_selected>false</denoise_selected>
@@ -18,11 +18,11 @@
     <service_provider_name nil="true"/>
     <timecode_source>embedded</timecode_source>
     <device_input>
-      <device_type>AJA</device_type>
-      <device_number>{{ device_number }}</device_number>
-      <channel>{{ channel }}</channel>
-      <channel_type>HD-SDI</channel_type>
-      <device_name>HD-SDI 1</device_name>
+      <device_type>{{ input_device["device_type"] }}</device_type>
+      <device_number>{{ input_device["device_number"] }}</device_number>
+      <channel>{{ input_device["channel"] }}</channel>
+      <channel_type>{{ input_device["channel_type"] }}</channel_type>
+      <device_name>{{ input_device["device_name"] }}</device_name>
       <name nil="true"/>
       <sdi_settings>
         <input_format>Auto</input_format>
@@ -835,7 +835,7 @@
       <destination>
         <password>{{ password }}</password>
         <username>{{ username }}</username>
-        <uri>{{ mediastore_container_master}}</uri>
+        <uri>{{ mediastore_container_master }}</uri>
       </destination>
     </apple_live_group_settings>
     <type>apple_live_group_settings</type>

setup.py

@@ -7,11 +7,12 @@
     name='python-elemental',
     author='CBS Interactive',
     author_email='[email protected]',
-    version='0.2',
+    version='0.4',
     include_patckage_data=True,
     url='https://github.com/cbsinteractive/elemental.git',
     license='MIT',
     long_description=open('README.md').read(),
     install_requires=requirements,
     packages=setuptools.find_packages(),
+    package_data={'elemental': ['templates/*']}
 )