ciocore 8.0.0b27__py2.py3-none-any.whl → 8.0.0b29__py2.py3-none-any.whl

ciocore/VERSION

@@ -1 +1 @@
-8.0.0-beta.27
+8.0.0-beta.29

ciocore/api_client.py

@@ -572,6 +572,38 @@ def request_software_packages():
     software = [sw for sw in software if not ("3dsmax" in sw["product"] and sw["platform"] == "linux")]
     return software
 
+def request_extra_environment():
+    """
+    Query Conductor for extra environment.
+    """
+    api = ApiClient()
+
+    uri = "api/v1/integrations/env-vars-configs"
+    response, response_code = api.make_request(
+        uri_path=uri, verb="GET", raise_on_error=False, use_api_key=True
+    )
+
+    if response_code not in [200]:
+        msg = "Failed to get extra environment"
+        msg += "\nError %s ...\n%s" % (response_code, response)
+        raise Exception(msg)
+
+    all_accounts = json.loads(response).get("data", [])
+
+    token = read_conductor_credentials(True)
+    if not token:
+        raise Exception("Error: Could not get conductor credentials!")
+    account_id =  str(account_id_from_jwt(token))
+
+    if not account_id:
+        raise Exception("Error: Could not get account id from jwt!")
+    account_env = next((account for account in all_accounts if account["account_id"] == account_id), None)
+    if not account_env:
+        raise Exception("Error: Could not get account environment!")
+    return account_env.get("env", [])
+
+
+
 
 def get_jobs(first_jid, last_jid=None):
     """

ciocore/cli.py

@@ -213,7 +213,7 @@ def upload(
     type=click.Choice(choices=PACKAGES_FORMAT_OPTIONS, case_sensitive=False),
 )
 def packages(fmt):
-    """List the softweare packages available on the render nodes in the cloud."""
+    """List the software packages available on the render nodes in the cloud."""
     package_query.pq(format=fmt)
 
 

ciocore/data.py

@@ -122,6 +122,14 @@ def data(force=False):
         # SOFTWARE
         software = api_client.request_software_packages()
 
+        # EXTRA ENV VARS
+        extra_env_vars = []
+        try:
+            extra_env_vars = api_client.request_extra_environment()
+        except Exception:
+            pass   
+        __data__["extra_environment"] = extra_env_vars
+
         # PLATFORMS
         it_platforms = set([it["operating_system"] for it in instance_types])
         valid_platforms = it_platforms.intersection(__platforms__)

ciocore/docsite/apidoc/api_client/index.html

@@ -674,6 +674,13 @@
     request_software_packages()
   </a>
   
+</li>
+      
+        <li class="md-nav__item">
+  <a href="#ciocore.api_client.request_extra_environment" class="md-nav__link">
+    request_extra_environment()
+  </a>
+  
 </li>
       
         <li class="md-nav__item">
@@ -981,6 +988,13 @@
     request_software_packages()
   </a>
   
+</li>
+      
+        <li class="md-nav__item">
+  <a href="#ciocore.api_client.request_extra_environment" class="md-nav__link">
+    request_extra_environment()
+  </a>
+  
 </li>
       
         <li class="md-nav__item">
@@ -2586,6 +2600,59 @@ prompt the user to log in. </p>
 
 
 
+<h2 id="ciocore.api_client.request_extra_environment" class="doc doc-heading">
+<code class="highlight language-python"><span class="n">request_extra_environment</span><span class="p">()</span></code>
+
+
+</h2>
+
+    <div class="doc doc-contents ">
+
+      <p>Query Conductor for extra environment.</p>
+
+        <details class="quote">
+          <summary>Source code in <code>ciocore/api_client.py</code></summary>
+          <div class="highlight"><pre><span></span><code><span class="k">def</span> <span class="nf">request_extra_environment</span><span class="p">():</span>
+<span class="w">    </span><span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">    Query Conductor for extra environment.</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+    <span class="n">api</span> <span class="o">=</span> <span class="n">ApiClient</span><span class="p">()</span>
+
+    <span class="n">uri</span> <span class="o">=</span> <span class="s2">&quot;api/v1/integrations/env-vars-configs&quot;</span>
+    <span class="n">response</span><span class="p">,</span> <span class="n">response_code</span> <span class="o">=</span> <span class="n">api</span><span class="o">.</span><span class="n">make_request</span><span class="p">(</span>
+        <span class="n">uri_path</span><span class="o">=</span><span class="n">uri</span><span class="p">,</span> <span class="n">verb</span><span class="o">=</span><span class="s2">&quot;GET&quot;</span><span class="p">,</span> <span class="n">raise_on_error</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span> <span class="n">use_api_key</span><span class="o">=</span><span class="kc">True</span>
+    <span class="p">)</span>
+
+    <span class="k">if</span> <span class="n">response_code</span> <span class="ow">not</span> <span class="ow">in</span> <span class="p">[</span><span class="mi">200</span><span class="p">]:</span>
+        <span class="n">msg</span> <span class="o">=</span> <span class="s2">&quot;Failed to get extra environment&quot;</span>
+        <span class="n">msg</span> <span class="o">+=</span> <span class="s2">&quot;</span><span class="se">\n</span><span class="s2">Error </span><span class="si">%s</span><span class="s2"> ...</span><span class="se">\n</span><span class="si">%s</span><span class="s2">&quot;</span> <span class="o">%</span> <span class="p">(</span><span class="n">response_code</span><span class="p">,</span> <span class="n">response</span><span class="p">)</span>
+        <span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="n">msg</span><span class="p">)</span>
+
+    <span class="n">all_accounts</span> <span class="o">=</span> <span class="n">json</span><span class="o">.</span><span class="n">loads</span><span class="p">(</span><span class="n">response</span><span class="p">)</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">&quot;data&quot;</span><span class="p">,</span> <span class="p">[])</span>
+
+    <span class="n">token</span> <span class="o">=</span> <span class="n">read_conductor_credentials</span><span class="p">(</span><span class="kc">True</span><span class="p">)</span>
+    <span class="k">if</span> <span class="ow">not</span> <span class="n">token</span><span class="p">:</span>
+        <span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">&quot;Error: Could not get conductor credentials!&quot;</span><span class="p">)</span>
+    <span class="n">account_id</span> <span class="o">=</span>  <span class="nb">str</span><span class="p">(</span><span class="n">account_id_from_jwt</span><span class="p">(</span><span class="n">token</span><span class="p">))</span>
+
+    <span class="k">if</span> <span class="ow">not</span> <span class="n">account_id</span><span class="p">:</span>
+        <span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">&quot;Error: Could not get account id from jwt!&quot;</span><span class="p">)</span>
+    <span class="n">account_env</span> <span class="o">=</span> <span class="nb">next</span><span class="p">((</span><span class="n">account</span> <span class="k">for</span> <span class="n">account</span> <span class="ow">in</span> <span class="n">all_accounts</span> <span class="k">if</span> <span class="n">account</span><span class="p">[</span><span class="s2">&quot;account_id&quot;</span><span class="p">]</span> <span class="o">==</span> <span class="n">account_id</span><span class="p">),</span> <span class="kc">None</span><span class="p">)</span>
+    <span class="k">if</span> <span class="ow">not</span> <span class="n">account_env</span><span class="p">:</span>
+        <span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">&quot;Error: Could not get account environment!&quot;</span><span class="p">)</span>
+    <span class="k">return</span> <span class="n">account_env</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">&quot;env&quot;</span><span class="p">,</span> <span class="p">[])</span>
+</code></pre></div>
+        </details>
+    </div>
+
+  </div>
+
+
+
+  <div class="doc doc-object doc-function">
+
+
+
 <h2 id="ciocore.api_client.get_jobs" class="doc doc-heading">
 <code class="highlight language-python"><span class="n">get_jobs</span><span class="p">(</span><span class="n">first_jid</span><span class="p">,</span> <span class="n">last_jid</span><span class="o">=</span><span class="kc">None</span><span class="p">)</span></code>
 

ciocore/docsite/apidoc/data/index.html

@@ -1150,6 +1150,14 @@ the software available at Conductor, or a subset based on specified products.</p
         <span class="c1"># SOFTWARE</span>
         <span class="n">software</span> <span class="o">=</span> <span class="n">api_client</span><span class="o">.</span><span class="n">request_software_packages</span><span class="p">()</span>
 
+        <span class="c1"># EXTRA ENV VARS</span>
+        <span class="n">extra_env_vars</span> <span class="o">=</span> <span class="p">[]</span>
+        <span class="k">try</span><span class="p">:</span>
+            <span class="n">extra_env_vars</span> <span class="o">=</span> <span class="n">api_client</span><span class="o">.</span><span class="n">request_extra_environment</span><span class="p">()</span>
+        <span class="k">except</span> <span class="ne">Exception</span><span class="p">:</span>
+            <span class="k">pass</span>   
+        <span class="n">__data__</span><span class="p">[</span><span class="s2">&quot;extra_environment&quot;</span><span class="p">]</span> <span class="o">=</span> <span class="n">extra_env_vars</span>
+
         <span class="c1"># PLATFORMS</span>
         <span class="n">it_platforms</span> <span class="o">=</span> <span class="nb">set</span><span class="p">([</span><span class="n">it</span><span class="p">[</span><span class="s2">&quot;operating_system&quot;</span><span class="p">]</span> <span class="k">for</span> <span class="n">it</span> <span class="ow">in</span> <span class="n">instance_types</span><span class="p">])</span>
         <span class="n">valid_platforms</span> <span class="o">=</span> <span class="n">it_platforms</span><span class="o">.</span><span class="n">intersection</span><span class="p">(</span><span class="n">__platforms__</span><span class="p">)</span>