mirror of
https://github.com/hakril/PythonForWindows
synced 2026-06-08 14:31:45 +00:00
502 lines
65 KiB
HTML
502 lines
65 KiB
HTML
<!DOCTYPE html>
|
|
|
|
<html lang="en" data-content_root="../../">
|
|
<head>
|
|
<meta charset="utf-8" />
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
|
<title>windows.com — PythonForWindows 1.0.2 documentation</title>
|
|
<link rel="stylesheet" type="text/css" href="../../_static/pygments.css?v=fa44fd50" />
|
|
<link rel="stylesheet" type="text/css" href="../../_static/classic.css?v=def86cc0" />
|
|
<link rel="stylesheet" type="text/css" href="../../_static/css/mbasic.css?v=957880af" />
|
|
|
|
<script src="../../_static/documentation_options.js?v=1ed6394b"></script>
|
|
<script src="../../_static/doctools.js?v=9a2dae69"></script>
|
|
<script src="../../_static/sphinx_highlight.js?v=dc90522c"></script>
|
|
|
|
<link rel="index" title="Index" href="../../genindex.html" />
|
|
<link rel="search" title="Search" href="../../search.html" />
|
|
</head><body>
|
|
<div class="related" role="navigation" aria-label="related navigation">
|
|
<h3>Navigation</h3>
|
|
<ul>
|
|
<li class="right" style="margin-right: 10px">
|
|
<a href="../../genindex.html" title="General Index"
|
|
accesskey="I">index</a></li>
|
|
<li class="right" >
|
|
<a href="../../py-modindex.html" title="Python Module Index"
|
|
>modules</a> |</li>
|
|
<li class="nav-item nav-item-0"><a href="../../index.html">PythonForWindows 1.0.2 documentation</a> »</li>
|
|
<li class="nav-item nav-item-1"><a href="../index.html" accesskey="U">Module code</a> »</li>
|
|
<li class="nav-item nav-item-this"><a href="">windows.com</a></li>
|
|
</ul>
|
|
</div>
|
|
|
|
<div class="document">
|
|
<div class="documentwrapper">
|
|
<div class="bodywrapper">
|
|
<div class="body" role="main">
|
|
|
|
<h1>Source code for windows.com</h1><div class="highlight"><pre>
|
|
<span></span><span class="kn">import</span> <span class="nn">sys</span>
|
|
<span class="kn">import</span> <span class="nn">struct</span>
|
|
<span class="kn">import</span> <span class="nn">ctypes</span>
|
|
<span class="kn">import</span> <span class="nn">functools</span>
|
|
<span class="kn">from</span> <span class="nn">ctypes</span> <span class="kn">import</span> <span class="n">HRESULT</span><span class="p">,</span> <span class="n">byref</span><span class="p">,</span> <span class="n">cast</span>
|
|
|
|
<span class="kn">import</span> <span class="nn">windows</span>
|
|
<span class="kn">from</span> <span class="nn">windows</span> <span class="kn">import</span> <span class="n">winproxy</span>
|
|
<span class="kn">from</span> <span class="nn">windows.generated_def.winstructs</span> <span class="kn">import</span> <span class="o">*</span>
|
|
|
|
<span class="kn">import</span> <span class="nn">windows.generated_def</span> <span class="k">as</span> <span class="nn">gdef</span>
|
|
<span class="kn">from</span> <span class="nn">windows.generated_def</span> <span class="kn">import</span> <span class="n">RPC_C_IMP_LEVEL_IMPERSONATE</span><span class="p">,</span> <span class="n">CLSCTX_INPROC_SERVER</span>
|
|
<span class="kn">from</span> <span class="nn">windows.generated_def</span> <span class="kn">import</span> <span class="n">interfaces</span>
|
|
<span class="kn">from</span> <span class="nn">windows.generated_def.interfaces</span> <span class="kn">import</span> <span class="n">generate_IID</span><span class="p">,</span> <span class="n">IID</span>
|
|
|
|
<span class="kn">from</span> <span class="nn">windows.pycompat</span> <span class="kn">import</span> <span class="n">int_types</span><span class="p">,</span> <span class="n">basestring</span>
|
|
|
|
<span class="c1"># We have windows.com.COMImplementation</span>
|
|
<span class="c1"># So we need windows.com.COMInterface</span>
|
|
<span class="n">COMInterface</span> <span class="o">=</span> <span class="n">interfaces</span><span class="o">.</span><span class="n">COMInterface</span>
|
|
|
|
<span class="c1"># Simple raw -> UUID</span>
|
|
<span class="c1"># "-".join("{:02X}".format(c) for c in struct.unpack("<IHHHBBBBBB", x))</span>
|
|
|
|
<div class="viewcode-block" id="init">
|
|
<a class="viewcode-back" href="../../com.html#windows.com.init">[docs]</a>
|
|
<span class="k">def</span> <span class="nf">init</span><span class="p">():</span>
|
|
<span class="w"> </span><span class="sd">"""Init COM with some default parameters"""</span>
|
|
<span class="k">try</span><span class="p">:</span>
|
|
<span class="n">t</span> <span class="o">=</span> <span class="n">winproxy</span><span class="o">.</span><span class="n">CoInitializeEx</span><span class="p">()</span>
|
|
<span class="k">except</span> <span class="ne">WindowsError</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
|
|
<span class="n">t</span> <span class="o">=</span> <span class="n">e</span><span class="o">.</span><span class="n">winerror</span>
|
|
<span class="k">if</span> <span class="n">t</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="n">t</span> <span class="o">&</span> <span class="mh">0xffffffff</span>
|
|
<span class="k">return</span> <span class="n">initsecurity</span><span class="p">()</span></div>
|
|
|
|
|
|
<span class="k">def</span> <span class="nf">initsecurity</span><span class="p">():</span> <span class="c1"># Should take some parameters..</span>
|
|
<span class="k">return</span> <span class="n">winproxy</span><span class="o">.</span><span class="n">CoInitializeSecurity</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="o">-</span><span class="mi">1</span><span class="p">,</span> <span class="kc">None</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="n">RPC_C_IMP_LEVEL_IMPERSONATE</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span><span class="mi">0</span><span class="p">,</span><span class="mi">0</span><span class="p">)</span>
|
|
|
|
|
|
<span class="k">class</span> <span class="nc">Dispatch</span><span class="p">(</span><span class="n">interfaces</span><span class="o">.</span><span class="n">IDispatch</span><span class="p">):</span>
|
|
<span class="k">def</span> <span class="nf">TypeInfoCount</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="n">count</span> <span class="o">=</span> <span class="n">gdef</span><span class="o">.</span><span class="n">UINT</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">GetTypeInfoCount</span><span class="p">(</span><span class="n">count</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">count</span>
|
|
|
|
<span class="k">def</span> <span class="nf">type_info</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">idx</span><span class="p">):</span>
|
|
<span class="n">type_info</span> <span class="o">=</span> <span class="n">TypeInfo</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">GetTypeInfo</span><span class="p">(</span><span class="n">idx</span><span class="p">,</span> <span class="mi">0</span><span class="p">,</span> <span class="n">type_info</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">type_info</span>
|
|
|
|
<span class="k">class</span> <span class="nc">TypeInfo</span><span class="p">(</span><span class="n">interfaces</span><span class="o">.</span><span class="n">ITypeInfo</span><span class="p">):</span>
|
|
<span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">idx</span><span class="p">):</span>
|
|
<span class="n">res</span> <span class="o">=</span> <span class="n">gdef</span><span class="o">.</span><span class="n">LPFUNCDESC</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">GetFuncDesc</span><span class="p">(</span><span class="n">idx</span><span class="p">,</span> <span class="n">res</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">res</span>
|
|
|
|
<span class="k">def</span> <span class="nf">attr</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="n">res</span> <span class="o">=</span> <span class="n">gdef</span><span class="o">.</span><span class="n">LPTYPEATTR</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">GetTypeAttr</span><span class="p">(</span><span class="n">res</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">res</span>
|
|
|
|
<span class="k">def</span> <span class="nf">names</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">memid</span><span class="p">):</span>
|
|
<span class="n">size</span> <span class="o">=</span> <span class="n">gdef</span><span class="o">.</span><span class="n">UINT</span><span class="p">()</span>
|
|
<span class="n">x</span> <span class="o">=</span> <span class="p">(</span><span class="n">gdef</span><span class="o">.</span><span class="n">BSTR</span> <span class="o">*</span> <span class="mi">10</span><span class="p">)(</span><span class="o">*</span><span class="nb">tuple</span><span class="p">(</span><span class="n">gdef</span><span class="o">.</span><span class="n">BSTR</span><span class="p">()</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="mi">10</span><span class="p">)))</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">GetNames</span><span class="p">(</span><span class="n">memid</span><span class="p">,</span> <span class="n">x</span><span class="p">,</span> <span class="mi">10</span><span class="p">,</span> <span class="n">size</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">x</span><span class="p">[:</span><span class="n">size</span><span class="o">.</span><span class="n">value</span><span class="p">]</span>
|
|
|
|
<span class="k">def</span> <span class="nf">docu</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="nb">id</span><span class="p">):</span>
|
|
<span class="n">res</span> <span class="o">=</span> <span class="n">gdef</span><span class="o">.</span><span class="n">BSTR</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">GetDocumentation</span><span class="p">(</span><span class="nb">id</span><span class="p">,</span> <span class="n">res</span><span class="p">,</span> <span class="kc">None</span><span class="p">,</span> <span class="kc">None</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">res</span>
|
|
|
|
|
|
<span class="k">class</span> <span class="nc">MemoryIStream</span><span class="p">(</span><span class="n">gdef</span><span class="o">.</span><span class="n">IStream</span><span class="p">):</span>
|
|
<span class="nd">@classmethod</span>
|
|
<span class="k">def</span> <span class="nf">create</span><span class="p">(</span><span class="bp">cls</span><span class="p">):</span>
|
|
<span class="bp">self</span> <span class="o">=</span> <span class="bp">cls</span><span class="p">()</span>
|
|
<span class="n">windows</span><span class="o">.</span><span class="n">winproxy</span><span class="o">.</span><span class="n">CreateStreamOnHGlobal</span><span class="p">(</span><span class="n">ppstm</span><span class="o">=</span><span class="bp">self</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="bp">self</span>
|
|
|
|
<div class="viewcode-block" id="create_instance">
|
|
<a class="viewcode-back" href="../../com.html#windows.com.create_instance">[docs]</a>
|
|
<span class="k">def</span> <span class="nf">create_instance</span><span class="p">(</span><span class="n">clsiid</span><span class="p">,</span> <span class="n">targetinterface</span><span class="p">,</span> <span class="n">custom_iid</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">context</span><span class="o">=</span><span class="n">CLSCTX_INPROC_SERVER</span> <span class="o">|</span> <span class="n">CLSCTX_LOCAL_SERVER</span><span class="p">):</span>
|
|
<span class="w"> </span><span class="sd">"""A simple wrapper around ``CoCreateInstance <https://msdn.microsoft.com/en-us/library/windows/desktop/ms686615(v=vs.85).aspx>``"""</span>
|
|
<span class="k">if</span> <span class="n">custom_iid</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="n">custom_iid</span> <span class="o">=</span> <span class="n">targetinterface</span><span class="o">.</span><span class="n">IID</span>
|
|
<span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">clsiid</span><span class="p">,</span> <span class="n">basestring</span><span class="p">):</span>
|
|
<span class="n">clsiid</span> <span class="o">=</span> <span class="n">IID</span><span class="o">.</span><span class="n">from_string</span><span class="p">(</span><span class="n">clsiid</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">custom_iid</span><span class="p">,</span> <span class="n">basestring</span><span class="p">):</span>
|
|
<span class="n">custom_iid</span> <span class="o">=</span> <span class="n">IID</span><span class="o">.</span><span class="n">from_string</span><span class="p">(</span><span class="n">custom_iid</span><span class="p">)</span>
|
|
<span class="n">winproxy</span><span class="o">.</span><span class="n">CoCreateInstance</span><span class="p">(</span><span class="n">byref</span><span class="p">(</span><span class="n">clsiid</span><span class="p">),</span> <span class="kc">None</span><span class="p">,</span> <span class="n">context</span><span class="p">,</span> <span class="n">byref</span><span class="p">(</span><span class="n">custom_iid</span><span class="p">),</span> <span class="n">byref</span><span class="p">(</span><span class="n">targetinterface</span><span class="p">))</span>
|
|
<span class="k">return</span> <span class="n">targetinterface</span></div>
|
|
|
|
|
|
|
|
<span class="k">def</span> <span class="nf">resolve_progid</span><span class="p">(</span><span class="n">progid</span><span class="p">):</span>
|
|
<span class="n">clsid</span> <span class="o">=</span> <span class="n">CLSID</span><span class="p">()</span>
|
|
<span class="n">winproxy</span><span class="o">.</span><span class="n">CLSIDFromProgID</span><span class="p">(</span><span class="n">progid</span><span class="p">,</span> <span class="n">clsid</span><span class="p">)</span>
|
|
<span class="c1"># We just filed the CLSID: refresh the __repr__</span>
|
|
<span class="n">clsid</span><span class="o">.</span><span class="n">update_strid</span><span class="p">()</span>
|
|
<span class="k">return</span> <span class="n">clsid</span>
|
|
|
|
<span class="c1"># Improved COM object</span>
|
|
<span class="c1"># Todo: ctypes_generation extended struct ?</span>
|
|
<span class="k">class</span> <span class="nc">SafeArray</span><span class="p">(</span><span class="n">SAFEARRAY</span><span class="p">):</span>
|
|
<span class="nd">@classmethod</span>
|
|
<span class="k">def</span> <span class="nf">create</span><span class="p">(</span><span class="bp">cls</span><span class="p">,</span> <span class="n">elttype</span><span class="p">,</span> <span class="n">dims</span><span class="p">):</span>
|
|
<span class="c1"># Array of bound for each dimension</span>
|
|
<span class="n">bound_array</span> <span class="o">=</span> <span class="n">windows</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">BUFFER</span><span class="p">(</span><span class="n">gdef</span><span class="o">.</span><span class="n">SAFEARRAYBOUND</span><span class="p">)(</span><span class="n">nbelt</span><span class="o">=</span><span class="nb">len</span><span class="p">(</span><span class="n">dims</span><span class="p">))</span>
|
|
<span class="k">for</span> <span class="n">i</span><span class="p">,</span> <span class="n">dim</span> <span class="ow">in</span> <span class="nb">enumerate</span><span class="p">(</span><span class="n">dims</span><span class="p">):</span>
|
|
<span class="n">bound_array</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">lLbound</span> <span class="o">=</span> <span class="mi">0</span>
|
|
<span class="n">bound_array</span><span class="p">[</span><span class="n">i</span><span class="p">]</span><span class="o">.</span><span class="n">cElements</span> <span class="o">=</span> <span class="n">dim</span>
|
|
<span class="n">array_ptr</span> <span class="o">=</span> <span class="n">windows</span><span class="o">.</span><span class="n">winproxy</span><span class="o">.</span><span class="n">SafeArrayCreate</span><span class="p">(</span><span class="n">elttype</span><span class="p">,</span> <span class="nb">len</span><span class="p">(</span><span class="n">dims</span><span class="p">),</span> <span class="n">bound_array</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="bp">cls</span><span class="o">.</span><span class="n">from_PSAFEARRAY</span><span class="p">(</span><span class="n">array_ptr</span><span class="p">)</span>
|
|
|
|
|
|
<span class="nd">@classmethod</span>
|
|
<span class="k">def</span> <span class="nf">of_type</span><span class="p">(</span><span class="bp">cls</span><span class="p">,</span> <span class="n">addr</span><span class="p">,</span> <span class="n">t</span><span class="p">):</span>
|
|
<span class="bp">self</span> <span class="o">=</span> <span class="bp">cls</span><span class="o">.</span><span class="n">from_address</span><span class="p">(</span><span class="n">addr</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">elt_type</span> <span class="o">=</span> <span class="n">t</span>
|
|
<span class="k">return</span> <span class="bp">self</span>
|
|
|
|
<span class="nd">@classmethod</span>
|
|
<span class="k">def</span> <span class="nf">from_PSAFEARRAY</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">psafearray</span><span class="p">):</span>
|
|
<span class="n">res</span> <span class="o">=</span> <span class="n">cast</span><span class="p">(</span><span class="n">psafearray</span><span class="p">,</span> <span class="n">POINTER</span><span class="p">(</span><span class="n">SafeArray</span><span class="p">))[</span><span class="mi">0</span><span class="p">]</span>
|
|
<span class="k">return</span> <span class="n">res</span>
|
|
|
|
<span class="k">def</span> <span class="nf">to_list</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">t</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">t</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="nb">hasattr</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="s2">"elt_type"</span><span class="p">):</span>
|
|
<span class="n">t</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">elt_type</span>
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Missing type of the array"</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">cDims</span> <span class="o">!=</span> <span class="mi">1</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="ne">NotImplementedError</span><span class="p">(</span><span class="s2">"tagSAFEARRAY if dims != 1"</span><span class="p">)</span>
|
|
|
|
<span class="n">nb_element</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">rgsabound</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">cElements</span>
|
|
<span class="n">llbound</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">rgsabound</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">lLbound</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">cbElements</span> <span class="o">!=</span> <span class="n">ctypes</span><span class="o">.</span><span class="n">sizeof</span><span class="p">(</span><span class="n">t</span><span class="p">):</span>
|
|
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Size of elements != sizeof(type)"</span><span class="p">)</span>
|
|
<span class="n">data</span> <span class="o">=</span> <span class="p">[</span><span class="n">t</span><span class="o">.</span><span class="n">from_address</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">pvData</span> <span class="o">+</span> <span class="p">(</span><span class="n">i</span> <span class="o">+</span> <span class="n">llbound</span><span class="p">)</span> <span class="o">*</span> <span class="n">ctypes</span><span class="o">.</span><span class="n">sizeof</span><span class="p">(</span><span class="n">t</span><span class="p">))</span><span class="o">.</span><span class="n">value</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">nb_element</span><span class="p">)]</span>
|
|
<span class="k">return</span> <span class="n">data</span>
|
|
|
|
<span class="k">def</span> <span class="nf">put_element</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">indexes</span><span class="p">,</span> <span class="n">element</span><span class="p">):</span>
|
|
<span class="n">indexes_array</span> <span class="o">=</span> <span class="n">windows</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">BUFFER</span><span class="p">(</span><span class="n">gdef</span><span class="o">.</span><span class="n">LONG</span><span class="p">)(</span><span class="o">*</span><span class="n">indexes</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">element</span><span class="p">,</span> <span class="n">VARIANT</span><span class="p">):</span>
|
|
<span class="k">pass</span>
|
|
<span class="c1"># element = Variant(element)</span>
|
|
<span class="c1"># import pdb; pdb.set_trace()</span>
|
|
<span class="c1"># element = ctypes.pointer(variant)</span>
|
|
<span class="c1"># print(hex(ctypes.addressof(element)))</span>
|
|
<span class="c1"># windows.utils.sprint(element)</span>
|
|
<span class="c1"># import pdb; pdb.set_trace()</span>
|
|
<span class="n">windows</span><span class="o">.</span><span class="n">winproxy</span><span class="o">.</span><span class="n">SafeArrayPutElement</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">indexes_array</span><span class="p">,</span> <span class="n">element</span><span class="p">)</span>
|
|
|
|
<span class="k">def</span> <span class="nf">get_element</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">indexes</span><span class="p">):</span>
|
|
<span class="n">indexes_array</span> <span class="o">=</span> <span class="n">windows</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">BUFFER</span><span class="p">(</span><span class="n">gdef</span><span class="o">.</span><span class="n">LONG</span><span class="p">)(</span><span class="o">*</span><span class="n">indexes</span><span class="p">)</span>
|
|
<span class="n">element</span> <span class="o">=</span> <span class="n">ctypes</span><span class="o">.</span><span class="n">create_string_buffer</span><span class="p">(</span><span class="mh">0x1000</span><span class="p">)</span>
|
|
<span class="n">windows</span><span class="o">.</span><span class="n">winproxy</span><span class="o">.</span><span class="n">SafeArrayGetElement</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">indexes_array</span><span class="p">,</span> <span class="n">element</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">element</span> <span class="c1"># Need cleaning ?</span>
|
|
|
|
<span class="k">def</span> <span class="fm">__repr__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="n">bound_array</span> <span class="o">=</span> <span class="n">windows</span><span class="o">.</span><span class="n">utils</span><span class="o">.</span><span class="n">resized_array</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">rgsabound</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">cDims</span><span class="p">)</span>
|
|
<span class="n">dims</span> <span class="o">=</span> <span class="p">[</span><span class="nb">int</span><span class="p">(</span><span class="n">b</span><span class="o">.</span><span class="n">cElements</span><span class="p">)</span> <span class="k">for</span> <span class="n">b</span> <span class="ow">in</span> <span class="n">bound_array</span><span class="p">]</span>
|
|
<span class="k">return</span> <span class="s2">"<</span><span class="si">{0}</span><span class="s2"> of dimension </span><span class="si">{1}</span><span class="s2">: </span><span class="si">{2}</span><span class="s2">>"</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">,</span> <span class="nb">len</span><span class="p">(</span><span class="n">dims</span><span class="p">),</span> <span class="n">dims</span><span class="p">)</span>
|
|
|
|
<span class="c1">#VT_VALUE_TO_TYPE = {</span>
|
|
<span class="c1">#VT_I2 : SHORT,</span>
|
|
<span class="c1">#VT_I4 : LONG,</span>
|
|
<span class="c1">#VT_BSTR : BSTR,</span>
|
|
<span class="c1">#VT_VARIANT : VARIANT,</span>
|
|
<span class="c1">#VT_UI1 : UCHAR,</span>
|
|
<span class="c1">#VT_UI2 : USHORT,</span>
|
|
<span class="c1">#VT_UI4 : DWORD,</span>
|
|
<span class="c1">#VT_I8 : LONGLONG,</span>
|
|
<span class="c1">#VT_UI8 : ULONG64,</span>
|
|
<span class="c1">#VT_INT : INT,</span>
|
|
<span class="c1">#VT_UINT : UINT,</span>
|
|
<span class="c1">#VT_HRESULT : HRESULT,</span>
|
|
<span class="c1">#VT_PTR : PVOID,</span>
|
|
<span class="c1">#VT_LPSTR : LPCSTR,</span>
|
|
<span class="c1">#VT_LPWSTR : LPWSTR,</span>
|
|
<span class="c1">#}</span>
|
|
|
|
<span class="c1"># VARIANT type checker</span>
|
|
<span class="c1"># Allow to guess a VARIANT_TYPE og a python value</span>
|
|
|
|
<span class="k">def</span> <span class="nf">never_match</span><span class="p">(</span><span class="n">value</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="kc">False</span>
|
|
|
|
<span class="k">def</span> <span class="nf">check_type_null</span><span class="p">(</span><span class="n">value</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="n">value</span> <span class="ow">is</span> <span class="kc">None</span>
|
|
|
|
<span class="k">def</span> <span class="nf">check_type_i4</span><span class="p">(</span><span class="n">value</span><span class="p">):</span>
|
|
<span class="c1"># 31 ? as we may want to keep sign :)</span>
|
|
<span class="k">return</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">value</span><span class="p">,</span> <span class="n">int_types</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span><span class="n">value</span><span class="p">)</span><span class="o">.</span><span class="n">bit_length</span><span class="p">()</span> <span class="o"><=</span> <span class="mi">32</span>
|
|
|
|
<span class="k">def</span> <span class="nf">check_type_i8</span><span class="p">(</span><span class="n">value</span><span class="p">):</span>
|
|
<span class="c1"># 63 ? as we may want to keep sign :)</span>
|
|
<span class="k">return</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">value</span><span class="p">,</span> <span class="n">int_types</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span><span class="n">value</span><span class="p">)</span><span class="o">.</span><span class="n">bit_length</span><span class="p">()</span> <span class="o"><=</span> <span class="mi">64</span>
|
|
|
|
<span class="k">def</span> <span class="nf">check_type_bstr</span><span class="p">(</span><span class="n">value</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">value</span><span class="p">,</span> <span class="n">basestring</span><span class="p">)</span>
|
|
|
|
<span class="k">def</span> <span class="nf">check_type_bool</span><span class="p">(</span><span class="n">value</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">value</span><span class="p">,</span> <span class="nb">bool</span><span class="p">)</span>
|
|
|
|
<span class="k">def</span> <span class="nf">check_type_safearray</span><span class="p">(</span><span class="n">value</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">value</span><span class="p">,</span> <span class="n">LPSAFEARRAY</span><span class="p">)</span>
|
|
|
|
<span class="k">def</span> <span class="nf">check_type_array</span><span class="p">(</span><span class="n">value</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="kc">True</span>
|
|
|
|
|
|
<span class="n">VARIAN_NAME_3_TYPE</span> <span class="o">=</span> <span class="p">[</span><span class="n">f</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> <span class="k">for</span> <span class="n">f</span> <span class="ow">in</span> <span class="n">VARIANT</span><span class="o">.</span><span class="n">_fields_</span> <span class="k">if</span> <span class="n">f</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> <span class="o">==</span> <span class="s2">"_VARIANT_NAME_3"</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span>
|
|
|
|
<span class="n">empty</span> <span class="o">=</span> <span class="nb">object</span><span class="p">()</span>
|
|
<span class="k">class</span> <span class="nc">Variant</span><span class="p">(</span><span class="n">VARIANT</span><span class="p">):</span>
|
|
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="o">=</span><span class="n">empty</span><span class="p">,</span> <span class="nb">type</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="nb">type</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">set_value_and_type</span><span class="p">(</span><span class="n">value</span><span class="p">,</span> <span class="nb">type</span><span class="p">)</span>
|
|
<span class="k">return</span>
|
|
<span class="k">elif</span> <span class="n">value</span> <span class="ow">is</span> <span class="n">empty</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vt</span> <span class="o">=</span> <span class="n">VT_EMPTY</span>
|
|
<span class="k">return</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">guess_type_and_set_value</span><span class="p">(</span><span class="n">value</span><span class="p">)</span>
|
|
|
|
<span class="c1"># Copy raw-ctypes fields which is a descriptor :)</span>
|
|
<span class="n">rawvt</span> <span class="o">=</span> <span class="n">VARIANT</span><span class="o">.</span><span class="n">vt</span>
|
|
|
|
<span class="c1"># Most of the value in the colunm[1]</span>
|
|
<span class="c1"># are attribute of the sub-union _VARIANT_NAME_3</span>
|
|
<span class="c1"># This union must be ctypes-anonymous for this code to works</span>
|
|
<span class="c1"># We want to access these directly from the VARIANT</span>
|
|
<span class="c1"># to allow custom descriptor for complexe type to be referenced here</span>
|
|
<span class="n">CHECK_TYPE</span> <span class="o">=</span> <span class="p">[</span>
|
|
<span class="c1"># Order is important</span>
|
|
<span class="c1"># as VT_I4 check may match VT_BOOL values</span>
|
|
<span class="c1"># VT_BOOL check must be before VT_I4 one</span>
|
|
<span class="p">(</span><span class="n">VT_BOOL</span><span class="p">,</span> <span class="s2">"boolVal"</span><span class="p">,</span> <span class="n">check_type_bool</span><span class="p">),</span>
|
|
<span class="p">(</span><span class="n">VT_I4</span><span class="p">,</span> <span class="s2">"lVal"</span><span class="p">,</span> <span class="n">check_type_i4</span><span class="p">),</span>
|
|
<span class="p">(</span><span class="n">VT_I8</span><span class="p">,</span> <span class="s2">"llVal"</span><span class="p">,</span> <span class="n">check_type_i8</span><span class="p">),</span>
|
|
<span class="p">(</span><span class="n">VT_BSTR</span><span class="p">,</span> <span class="s2">"bstrVal"</span><span class="p">,</span> <span class="n">check_type_bstr</span><span class="p">),</span>
|
|
<span class="p">(</span><span class="n">VT_SAFEARRAY</span> <span class="p">,</span> <span class="s2">"parray"</span><span class="p">,</span> <span class="n">check_type_safearray</span><span class="p">),</span>
|
|
<span class="p">(</span><span class="n">VT_NULL</span><span class="p">,</span> <span class="kc">None</span><span class="p">,</span> <span class="n">check_type_null</span><span class="p">),</span>
|
|
<span class="p">(</span><span class="n">VT_EMPTY</span><span class="p">,</span> <span class="kc">None</span><span class="p">,</span> <span class="n">never_match</span><span class="p">),</span>
|
|
<span class="p">(</span><span class="n">VT_DISPATCH</span><span class="p">,</span> <span class="s2">"pdispVal"</span><span class="p">,</span> <span class="n">never_match</span><span class="p">),</span> <span class="c1"># I cannot recognize DISPATCH ptr for now</span>
|
|
<span class="p">(</span><span class="n">VT_UNKNOWN</span><span class="p">,</span> <span class="s2">"punkVal"</span><span class="p">,</span> <span class="n">never_match</span><span class="p">),</span> <span class="c1"># recognise PFW ComInterface ?</span>
|
|
<span class="c1"># Test: do not allow auto-creation of small int values</span>
|
|
<span class="c1"># I don't know but a feel it may confuse some API expecting VT_I4</span>
|
|
<span class="p">(</span><span class="n">VT_I2</span><span class="p">,</span> <span class="s2">"iVal"</span><span class="p">,</span> <span class="n">never_match</span><span class="p">),</span>
|
|
<span class="p">(</span><span class="n">VT_UI1</span><span class="p">,</span> <span class="s2">"bVal"</span><span class="p">,</span> <span class="n">never_match</span><span class="p">),</span>
|
|
<span class="p">]</span>
|
|
|
|
<span class="n">VARIANT_TYPE_BY_NAME</span> <span class="o">=</span> <span class="p">{</span><span class="n">f</span><span class="p">[</span><span class="mi">0</span><span class="p">]:</span> <span class="n">f</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span> <span class="k">for</span> <span class="n">f</span> <span class="ow">in</span> <span class="n">VARIAN_NAME_3_TYPE</span><span class="o">.</span><span class="n">_fields_</span><span class="p">}</span>
|
|
<span class="n">QUICK_CHECK_TYPE</span> <span class="o">=</span> <span class="p">{</span><span class="n">x</span><span class="p">:</span> <span class="n">y</span> <span class="k">for</span> <span class="n">x</span><span class="p">,</span><span class="n">y</span><span class="p">,</span> <span class="n">_</span> <span class="ow">in</span> <span class="n">CHECK_TYPE</span><span class="p">}</span>
|
|
|
|
<span class="k">def</span> <span class="nf">get_vt</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="n">rawvt</span> <span class="o">=</span> <span class="nb">super</span><span class="p">(</span><span class="n">Variant</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="n">vt</span>
|
|
<span class="k">return</span> <span class="n">gdef</span><span class="o">.</span><span class="n">VARENUM</span><span class="o">.</span><span class="n">mapper</span><span class="p">[</span><span class="bp">self</span><span class="o">.</span><span class="n">rawvt</span><span class="p">]</span>
|
|
|
|
<span class="k">def</span> <span class="nf">set_vt</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">rawvt</span> <span class="o">=</span> <span class="n">value</span>
|
|
|
|
<span class="n">vt</span> <span class="o">=</span> <span class="nb">property</span><span class="p">(</span><span class="n">get_vt</span><span class="p">,</span> <span class="n">set_vt</span><span class="p">)</span>
|
|
|
|
<span class="k">def</span> <span class="nf">set_value_and_type</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">,</span> <span class="nb">type</span><span class="p">):</span>
|
|
<span class="n">attr</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">QUICK_CHECK_TYPE</span><span class="p">[</span><span class="nb">type</span><span class="p">]</span>
|
|
<span class="c1"># No check: user must be careful about non-match value&type</span>
|
|
<span class="nb">setattr</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">attr</span><span class="p">,</span> <span class="n">value</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vt</span> <span class="o">=</span> <span class="nb">type</span>
|
|
|
|
<span class="k">def</span> <span class="nf">get_value_based_on_type</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="n">rawvt</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">rawvt</span>
|
|
<span class="k">if</span> <span class="n">rawvt</span> <span class="o">&</span> <span class="n">VT_ARRAY</span><span class="p">:</span>
|
|
<span class="n">realtype</span> <span class="o">=</span> <span class="n">rawvt</span> <span class="o">&</span> <span class="o">~</span><span class="n">VT_ARRAY</span>
|
|
<span class="n">attr</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">QUICK_CHECK_TYPE</span><span class="p">[</span><span class="n">realtype</span><span class="p">]</span>
|
|
<span class="n">attrtype</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">VARIANT_TYPE_BY_NAME</span><span class="p">[</span><span class="n">attr</span><span class="p">]</span>
|
|
<span class="n">array</span> <span class="o">=</span> <span class="n">SafeArray</span><span class="o">.</span><span class="n">from_PSAFEARRAY</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_VARIANT_NAME_3</span><span class="o">.</span><span class="n">parray</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">array</span><span class="o">.</span><span class="n">to_list</span><span class="p">(</span><span class="n">attrtype</span><span class="p">)</span>
|
|
<span class="n">attr</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">QUICK_CHECK_TYPE</span><span class="p">[</span><span class="n">rawvt</span><span class="p">]</span>
|
|
<span class="k">if</span> <span class="n">attr</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="kc">None</span>
|
|
<span class="k">if</span> <span class="n">attr</span> <span class="o">==</span> <span class="s2">"punkVal"</span><span class="p">:</span>
|
|
<span class="c1"># Quick hack for COM interface type</span>
|
|
<span class="c1"># Do something clean with CHECK_TYPE ?</span>
|
|
<span class="n">x</span> <span class="o">=</span> <span class="n">gdef</span><span class="o">.</span><span class="n">IUnknown</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">punkVal</span><span class="p">)</span>
|
|
<span class="n">x</span><span class="o">.</span><span class="n">AddRef</span><span class="p">()</span>
|
|
<span class="k">return</span> <span class="n">x</span>
|
|
<span class="k">return</span> <span class="nb">getattr</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">attr</span><span class="p">)</span>
|
|
|
|
<span class="k">def</span> <span class="nf">guess_type_and_set_value</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
|
|
<span class="k">for</span> <span class="n">t</span><span class="p">,</span> <span class="n">attr</span><span class="p">,</span> <span class="n">check</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">CHECK_TYPE</span><span class="p">:</span>
|
|
<span class="k">try</span><span class="p">:</span>
|
|
<span class="n">checkres</span> <span class="o">=</span> <span class="n">check</span><span class="p">(</span><span class="n">value</span><span class="p">)</span>
|
|
<span class="k">except</span> <span class="ne">TypeError</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
|
|
<span class="k">continue</span>
|
|
<span class="k">if</span> <span class="n">checkres</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vt</span> <span class="o">=</span> <span class="n">t</span>
|
|
<span class="k">if</span> <span class="n">attr</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="nb">setattr</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">attr</span><span class="p">,</span> <span class="n">value</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="kc">True</span>
|
|
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Could not guess VT_TYPE for <</span><span class="si">{0}</span><span class="s2">> of type <</span><span class="si">{1}</span><span class="s2">>"</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">value</span><span class="p">,</span> <span class="nb">type</span><span class="p">(</span><span class="n">value</span><span class="p">)))</span>
|
|
|
|
<span class="n">value</span> <span class="o">=</span> <span class="nb">property</span><span class="p">(</span><span class="n">get_value_based_on_type</span><span class="p">,</span> <span class="n">guess_type_and_set_value</span><span class="p">)</span>
|
|
|
|
<span class="c1"># quick_check: bypass python lookup-limitation</span>
|
|
<span class="k">def</span> <span class="nf">generate_getter</span><span class="p">(</span><span class="n">vt_type</span><span class="p">,</span> <span class="n">transfo</span><span class="o">=</span><span class="p">(</span><span class="k">lambda</span> <span class="n">x</span><span class="p">:</span><span class="n">x</span><span class="p">),</span> <span class="n">quick_check</span><span class="o">=</span><span class="n">QUICK_CHECK_TYPE</span><span class="p">):</span>
|
|
<span class="n">attr</span> <span class="o">=</span> <span class="n">quick_check</span><span class="p">[</span><span class="n">vt_type</span><span class="p">]</span>
|
|
<span class="nd">@property</span>
|
|
<span class="k">def</span> <span class="nf">getter</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">rawvt</span> <span class="o">==</span> <span class="n">vt_type</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Invalid vt-type for attribute expected <</span><span class="si">{0}</span><span class="s2">> got <</span><span class="si">{1}</span><span class="s2">>"</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">vt_type</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">vt</span><span class="p">))</span>
|
|
<span class="k">return</span> <span class="n">transfo</span><span class="p">(</span><span class="nb">getattr</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">attr</span><span class="p">))</span>
|
|
<span class="k">return</span> <span class="n">getter</span>
|
|
|
|
<span class="n">asbstr</span> <span class="o">=</span> <span class="n">generate_getter</span><span class="p">(</span><span class="n">VT_BSTR</span><span class="p">)</span>
|
|
<span class="n">aslong</span> <span class="o">=</span> <span class="n">generate_getter</span><span class="p">(</span><span class="n">VT_I4</span><span class="p">)</span>
|
|
<span class="n">asbool</span> <span class="o">=</span> <span class="n">generate_getter</span><span class="p">(</span><span class="n">VT_BOOL</span><span class="p">)</span>
|
|
<span class="n">asdispatch</span> <span class="o">=</span> <span class="n">generate_getter</span><span class="p">(</span><span class="n">VT_DISPATCH</span><span class="p">,</span> <span class="n">transfo</span><span class="o">=</span><span class="n">interfaces</span><span class="o">.</span><span class="n">IDispatch</span><span class="p">)</span>
|
|
<span class="n">asshort</span> <span class="o">=</span> <span class="n">generate_getter</span><span class="p">(</span><span class="n">VT_I2</span><span class="p">)</span>
|
|
<span class="n">asbyte</span> <span class="o">=</span> <span class="n">generate_getter</span><span class="p">(</span><span class="n">VT_UI1</span><span class="p">)</span>
|
|
<span class="n">asunknown</span> <span class="o">=</span> <span class="n">generate_getter</span><span class="p">(</span><span class="n">VT_UNKNOWN</span><span class="p">)</span>
|
|
|
|
<span class="k">def</span> <span class="fm">__repr__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="s2">"""<</span><span class="si">{0}</span><span class="s2"> of type </span><span class="si">{1}</span><span class="s2">>"""</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">vt</span><span class="p">)</span>
|
|
|
|
<span class="c1"># Deprecated: remove me when test pass :)</span>
|
|
<span class="c1"># ImprovedVariant.MAPPER = {</span>
|
|
<span class="c1"># VT_UI1: ImprovedVariant.asbyte.fget,</span>
|
|
<span class="c1"># VT_I2: ImprovedVariant.asshort.fget,</span>
|
|
<span class="c1"># VT_DISPATCH: ImprovedVariant.asdispatch.fget,</span>
|
|
<span class="c1"># VT_BOOL: ImprovedVariant.asbool.fget,</span>
|
|
<span class="c1"># VT_I4: ImprovedVariant.aslong.fget,</span>
|
|
<span class="c1"># VT_BSTR: ImprovedVariant.asbstr.fget,</span>
|
|
<span class="c1"># VT_EMPTY: (lambda x: None),</span>
|
|
<span class="c1"># VT_NULL: (lambda x: None),</span>
|
|
<span class="c1"># VT_UNKNOWN: ImprovedVariant.asunknown.fget,</span>
|
|
<span class="c1"># (VT_ARRAY | VT_BSTR): ImprovedVariant.asbstr_array.fget,</span>
|
|
<span class="c1"># (VT_ARRAY | VT_I4): ImprovedVariant.aslong_array.fget,</span>
|
|
<span class="c1"># (VT_ARRAY | VT_UI1): ImprovedVariant.asbyte_array.fget,</span>
|
|
<span class="c1"># (VT_ARRAY | VT_BOOL): ImprovedVariant.asbool_array.fget</span>
|
|
<span class="c1"># }</span>
|
|
|
|
|
|
|
|
<div class="viewcode-block" id="COMImplementation">
|
|
<a class="viewcode-back" href="../../com.html#windows.com.COMImplementation">[docs]</a>
|
|
<span class="k">class</span> <span class="nc">COMImplementation</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
|
|
<span class="w"> </span><span class="sd">"""The base class to implements COM object respecting a given interface"""</span>
|
|
<span class="n">IMPLEMENT</span> <span class="o">=</span> <span class="kc">None</span>
|
|
|
|
<span class="k">def</span> <span class="nf">get_index_of_method</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">method</span><span class="p">):</span>
|
|
<span class="c1"># This code is horrible but not totally my fault</span>
|
|
<span class="c1"># the PyCFuncPtrObject->index is not exposed to Python..</span>
|
|
<span class="c1"># repr is: '<COM method offset 2: WinFunctionType at 0x035DDBE8>'</span>
|
|
<span class="n">rpr</span> <span class="o">=</span> <span class="nb">repr</span><span class="p">(</span><span class="n">method</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">rpr</span><span class="o">.</span><span class="n">startswith</span><span class="p">(</span><span class="s2">"<COM method offset "</span><span class="p">)</span> <span class="ow">or</span> <span class="s2">":"</span> <span class="ow">not</span> <span class="ow">in</span> <span class="n">rpr</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Could not extract offset of </span><span class="si">{0}</span><span class="s2">"</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">rpr</span><span class="p">))</span>
|
|
<span class="k">return</span> <span class="nb">int</span><span class="p">(</span><span class="n">rpr</span><span class="p">[</span><span class="nb">len</span><span class="p">(</span><span class="s2">"<COM method offset "</span><span class="p">):</span> <span class="n">rpr</span><span class="o">.</span><span class="n">index</span><span class="p">(</span><span class="s2">":"</span><span class="p">)])</span>
|
|
|
|
<span class="k">def</span> <span class="nf">extract_methods_order</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">interface</span><span class="p">):</span>
|
|
<span class="n">index_and_method</span> <span class="o">=</span> <span class="nb">sorted</span><span class="p">((</span><span class="bp">self</span><span class="o">.</span><span class="n">get_index_of_method</span><span class="p">(</span><span class="n">m</span><span class="p">),</span><span class="n">name</span><span class="p">,</span> <span class="n">m</span><span class="p">)</span> <span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">m</span> <span class="ow">in</span> <span class="n">interface</span><span class="o">.</span><span class="n">_functions_</span><span class="o">.</span><span class="n">items</span><span class="p">())</span>
|
|
<span class="k">return</span> <span class="n">index_and_method</span>
|
|
|
|
<span class="k">def</span> <span class="nf">verify_implem</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">interface</span><span class="p">):</span>
|
|
<span class="k">for</span> <span class="n">func_name</span> <span class="ow">in</span> <span class="n">interface</span><span class="o">.</span><span class="n">_functions_</span><span class="p">:</span>
|
|
<span class="n">implem</span> <span class="o">=</span> <span class="nb">getattr</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">func_name</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">implem</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"<</span><span class="si">{0}</span><span class="s2">> implementing <</span><span class="si">{1}</span><span class="s2">> has no method <</span><span class="si">{2}</span><span class="s2">>"</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">IMPLEMENT</span><span class="o">.</span><span class="vm">__name__</span><span class="p">,</span> <span class="n">func_name</span><span class="p">))</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="nb">callable</span><span class="p">(</span><span class="n">implem</span><span class="p">):</span>
|
|
<span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"</span><span class="si">{0}</span><span class="s2"> implementing <</span><span class="si">{1}</span><span class="s2">>: <</span><span class="si">{2}</span><span class="s2">> is not callable"</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="vm">__name__</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">IMPLEMENT</span><span class="o">.</span><span class="vm">__name__</span><span class="p">,</span> <span class="n">func_name</span><span class="p">))</span>
|
|
<span class="k">return</span> <span class="kc">True</span>
|
|
|
|
<span class="k">def</span> <span class="nf">_create_vtable</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">interface</span><span class="p">):</span>
|
|
<span class="n">implems</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="n">names</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">index</span><span class="p">,</span> <span class="n">name</span><span class="p">,</span> <span class="n">method</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">extract_methods_order</span><span class="p">(</span><span class="n">interface</span><span class="p">):</span>
|
|
<span class="n">func_implem</span> <span class="o">=</span> <span class="nb">getattr</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">name</span><span class="p">)</span>
|
|
<span class="c1">#'this' is a COM-interface of the type we are implementing</span>
|
|
<span class="n">types</span> <span class="o">=</span> <span class="p">[</span><span class="n">method</span><span class="o">.</span><span class="n">restype</span><span class="p">,</span> <span class="n">interface</span><span class="p">]</span> <span class="o">+</span> <span class="nb">list</span><span class="p">(</span><span class="n">method</span><span class="o">.</span><span class="n">argtypes</span><span class="p">)</span>
|
|
<span class="n">implems</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">ctypes</span><span class="o">.</span><span class="n">WINFUNCTYPE</span><span class="p">(</span><span class="o">*</span><span class="n">types</span><span class="p">)(</span><span class="n">func_implem</span><span class="p">))</span>
|
|
<span class="n">names</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">name</span><span class="p">)</span>
|
|
<span class="k">class</span> <span class="nc">Vtable</span><span class="p">(</span><span class="n">ctypes</span><span class="o">.</span><span class="n">Structure</span><span class="p">):</span>
|
|
<span class="n">_fields_</span> <span class="o">=</span> <span class="p">[(</span><span class="n">name</span><span class="p">,</span> <span class="n">ctypes</span><span class="o">.</span><span class="n">c_void_p</span><span class="p">)</span> <span class="k">for</span> <span class="n">name</span> <span class="ow">in</span> <span class="n">names</span><span class="p">]</span>
|
|
<span class="k">return</span> <span class="n">Vtable</span><span class="p">(</span><span class="o">*</span><span class="p">[</span><span class="n">ctypes</span><span class="o">.</span><span class="n">cast</span><span class="p">(</span><span class="n">x</span><span class="p">,</span> <span class="n">ctypes</span><span class="o">.</span><span class="n">c_void_p</span><span class="p">)</span> <span class="k">for</span> <span class="n">x</span> <span class="ow">in</span> <span class="n">implems</span><span class="p">]),</span> <span class="n">implems</span>
|
|
|
|
<span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">verify_implem</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">IMPLEMENT</span><span class="p">)</span>
|
|
<span class="n">vtable</span><span class="p">,</span> <span class="n">implems</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_create_vtable</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">IMPLEMENT</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vtable</span> <span class="o">=</span> <span class="n">vtable</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">implems</span> <span class="o">=</span> <span class="n">implems</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vtable_pointer</span> <span class="o">=</span> <span class="n">ctypes</span><span class="o">.</span><span class="n">pointer</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">vtable</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_as_parameter_</span> <span class="o">=</span> <span class="n">ctypes</span><span class="o">.</span><span class="n">addressof</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">vtable_pointer</span><span class="p">)</span>
|
|
|
|
<div class="viewcode-block" id="COMImplementation.QueryInterface">
|
|
<a class="viewcode-back" href="../../com.html#windows.com.COMImplementation.QueryInterface">[docs]</a>
|
|
<span class="k">def</span> <span class="nf">QueryInterface</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">this</span><span class="p">,</span> <span class="n">piid</span><span class="p">,</span> <span class="n">result</span><span class="p">):</span>
|
|
<span class="w"> </span><span class="sd">"""Default ``QueryInterface`` implementation that returns ``self`` if piid is the implemented interface"""</span>
|
|
<span class="k">if</span> <span class="n">piid</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> <span class="ow">in</span> <span class="p">(</span><span class="n">gdef</span><span class="o">.</span><span class="n">IUnknown</span><span class="o">.</span><span class="n">IID</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">IMPLEMENT</span><span class="o">.</span><span class="n">IID</span><span class="p">):</span>
|
|
<span class="n">result</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span> <span class="o">=</span> <span class="n">this</span>
|
|
<span class="k">return</span> <span class="mi">1</span>
|
|
<span class="k">return</span> <span class="n">E_NOINTERFACE</span></div>
|
|
|
|
|
|
<div class="viewcode-block" id="COMImplementation.AddRef">
|
|
<a class="viewcode-back" href="../../com.html#windows.com.COMImplementation.AddRef">[docs]</a>
|
|
<span class="k">def</span> <span class="nf">AddRef</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="o">*</span><span class="n">args</span><span class="p">):</span>
|
|
<span class="w"> </span><span class="sd">"""Default ``AddRef`` implementation that returns ``1``"""</span>
|
|
<span class="k">return</span> <span class="mi">1</span></div>
|
|
|
|
|
|
<div class="viewcode-block" id="COMImplementation.Release">
|
|
<a class="viewcode-back" href="../../com.html#windows.com.COMImplementation.Release">[docs]</a>
|
|
<span class="k">def</span> <span class="nf">Release</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="o">*</span><span class="n">args</span><span class="p">):</span>
|
|
<span class="w"> </span><span class="sd">"""Default ``Release`` implementation that returns ``1``"""</span>
|
|
<span class="k">return</span> <span class="mi">0</span></div>
|
|
</div>
|
|
|
|
</pre></div>
|
|
|
|
<div class="clearer"></div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
<div class="sphinxsidebar" role="navigation" aria-label="main navigation">
|
|
<div class="sphinxsidebarwrapper">
|
|
<search id="searchbox" style="display: none" role="search">
|
|
<h3 id="searchlabel">Quick search</h3>
|
|
<div class="searchformwrapper">
|
|
<form class="search" action="../../search.html" method="get">
|
|
<input type="text" name="q" aria-labelledby="searchlabel" autocomplete="off" autocorrect="off" autocapitalize="off" spellcheck="false"/>
|
|
<input type="submit" value="Go" />
|
|
</form>
|
|
</div>
|
|
</search>
|
|
<script>document.getElementById('searchbox').style.display = "block"</script>
|
|
</div>
|
|
</div>
|
|
<div class="clearer"></div>
|
|
</div>
|
|
<div class="related" role="navigation" aria-label="related navigation">
|
|
<h3>Navigation</h3>
|
|
<ul>
|
|
<li class="right" style="margin-right: 10px">
|
|
<a href="../../genindex.html" title="General Index"
|
|
>index</a></li>
|
|
<li class="right" >
|
|
<a href="../../py-modindex.html" title="Python Module Index"
|
|
>modules</a> |</li>
|
|
<li class="nav-item nav-item-0"><a href="../../index.html">PythonForWindows 1.0.2 documentation</a> »</li>
|
|
<li class="nav-item nav-item-1"><a href="../index.html" >Module code</a> »</li>
|
|
<li class="nav-item nav-item-this"><a href="">windows.com</a></li>
|
|
</ul>
|
|
</div>
|
|
<div class="footer" role="contentinfo">
|
|
© Copyright 2015-2020, Clement Rouault.
|
|
Created using <a href="https://www.sphinx-doc.org/">Sphinx</a> 7.3.7.
|
|
</div>
|
|
</body>
|
|
</html> |