A clone of: https://github.com/nutechsoftware/alarmdecoder This is requires as they dropped support for older firmware releases w/o building in backward compatibility code, and they had previously hardcoded pyserial to a python2 only version.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1008 lines
96 KiB

  1. <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
  2. "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
  3. <html xmlns="http://www.w3.org/1999/xhtml">
  4. <head>
  5. <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
  6. <title>pyad2.devices &mdash; pyad2 documentation</title>
  7. <link rel="stylesheet" href="../../_static/default.css" type="text/css" />
  8. <link rel="stylesheet" href="../../_static/pygments.css" type="text/css" />
  9. <script type="text/javascript">
  10. var DOCUMENTATION_OPTIONS = {
  11. URL_ROOT: '../../',
  12. VERSION: '',
  13. COLLAPSE_INDEX: false,
  14. FILE_SUFFIX: '.html',
  15. HAS_SOURCE: true
  16. };
  17. </script>
  18. <script type="text/javascript" src="../../_static/jquery.js"></script>
  19. <script type="text/javascript" src="../../_static/underscore.js"></script>
  20. <script type="text/javascript" src="../../_static/doctools.js"></script>
  21. <link rel="top" title="pyad2 documentation" href="../../index.html" />
  22. <link rel="up" title="Module code" href="../index.html" />
  23. </head>
  24. <body>
  25. <div class="related">
  26. <h3>Navigation</h3>
  27. <ul>
  28. <li class="right" style="margin-right: 10px">
  29. <a href="../../genindex.html" title="General Index"
  30. accesskey="I">index</a></li>
  31. <li class="right" >
  32. <a href="../../py-modindex.html" title="Python Module Index"
  33. >modules</a> |</li>
  34. <li><a href="../../index.html">pyad2 documentation</a> &raquo;</li>
  35. <li><a href="../index.html" accesskey="U">Module code</a> &raquo;</li>
  36. </ul>
  37. </div>
  38. <div class="document">
  39. <div class="documentwrapper">
  40. <div class="bodywrapper">
  41. <div class="body">
  42. <h1>Source code for pyad2.devices</h1><div class="highlight"><pre>
  43. <span class="sd">&quot;&quot;&quot;</span>
  44. <span class="sd">Contains different types of devices belonging to the AD2 family.</span>
  45. <span class="sd">.. moduleauthor:: Scott Petersen &lt;scott@nutech.com&gt;</span>
  46. <span class="sd">&quot;&quot;&quot;</span>
  47. <span class="kn">import</span> <span class="nn">usb.core</span><span class="o">,</span> <span class="nn">usb.util</span>
  48. <span class="kn">import</span> <span class="nn">time</span>
  49. <span class="kn">import</span> <span class="nn">threading</span>
  50. <span class="kn">import</span> <span class="nn">serial</span><span class="o">,</span> <span class="nn">serial.tools.list_ports</span>
  51. <span class="kn">import</span> <span class="nn">socket</span>
  52. <span class="kn">from</span> <span class="nn">OpenSSL</span> <span class="kn">import</span> <span class="n">SSL</span><span class="p">,</span> <span class="n">crypto</span>
  53. <span class="kn">from</span> <span class="nn">pyftdi.pyftdi.ftdi</span> <span class="kn">import</span> <span class="o">*</span>
  54. <span class="kn">from</span> <span class="nn">pyftdi.pyftdi.usbtools</span> <span class="kn">import</span> <span class="o">*</span>
  55. <span class="kn">from</span> <span class="nn">.util</span> <span class="kn">import</span> <span class="n">CommError</span><span class="p">,</span> <span class="n">TimeoutError</span><span class="p">,</span> <span class="n">NoDeviceError</span>
  56. <span class="kn">from</span> <span class="nn">.event</span> <span class="kn">import</span> <span class="n">event</span>
  57. <div class="viewcode-block" id="Device"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.Device">[docs]</a><span class="k">class</span> <span class="nc">Device</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
  58. <span class="sd">&quot;&quot;&quot;</span>
  59. <span class="sd"> Generic parent device to all AD2 products.</span>
  60. <span class="sd"> &quot;&quot;&quot;</span>
  61. <span class="c"># Generic device events</span>
  62. <span class="n">on_open</span> <span class="o">=</span> <span class="n">event</span><span class="o">.</span><span class="n">Event</span><span class="p">(</span><span class="s">&#39;Called when the device has been opened&#39;</span><span class="p">)</span>
  63. <span class="n">on_close</span> <span class="o">=</span> <span class="n">event</span><span class="o">.</span><span class="n">Event</span><span class="p">(</span><span class="s">&#39;Called when the device has been closed&#39;</span><span class="p">)</span>
  64. <span class="n">on_read</span> <span class="o">=</span> <span class="n">event</span><span class="o">.</span><span class="n">Event</span><span class="p">(</span><span class="s">&#39;Called when a line has been read from the device&#39;</span><span class="p">)</span>
  65. <span class="n">on_write</span> <span class="o">=</span> <span class="n">event</span><span class="o">.</span><span class="n">Event</span><span class="p">(</span><span class="s">&#39;Called when data has been written to the device&#39;</span><span class="p">)</span>
  66. <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  67. <span class="sd">&quot;&quot;&quot;</span>
  68. <span class="sd"> Constructor</span>
  69. <span class="sd"> &quot;&quot;&quot;</span>
  70. <span class="bp">self</span><span class="o">.</span><span class="n">_id</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>
  71. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>
  72. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span> <span class="o">=</span> <span class="bp">None</span>
  73. <span class="bp">self</span><span class="o">.</span><span class="n">_running</span> <span class="o">=</span> <span class="bp">False</span>
  74. <span class="bp">self</span><span class="o">.</span><span class="n">_read_thread</span> <span class="o">=</span> <span class="n">Device</span><span class="o">.</span><span class="n">ReadThread</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
  75. <span class="nd">@property</span>
  76. <span class="k">def</span> <span class="nf">id</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  77. <span class="sd">&quot;&quot;&quot;</span>
  78. <span class="sd"> Retrieve the device ID.</span>
  79. <span class="sd"> :returns: The identification string for the device.</span>
  80. <span class="sd"> &quot;&quot;&quot;</span>
  81. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_id</span>
  82. <span class="nd">@id.setter</span>
  83. <div class="viewcode-block" id="Device.id"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.Device.id">[docs]</a> <span class="k">def</span> <span class="nf">id</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  84. <span class="sd">&quot;&quot;&quot;</span>
  85. <span class="sd"> Sets the device ID.</span>
  86. <span class="sd"> :param value: The device identification.</span>
  87. <span class="sd"> :type value: str</span>
  88. <span class="sd"> &quot;&quot;&quot;</span>
  89. <span class="bp">self</span><span class="o">.</span><span class="n">_id</span> <span class="o">=</span> <span class="n">value</span>
  90. </div>
  91. <div class="viewcode-block" id="Device.is_reader_alive"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.Device.is_reader_alive">[docs]</a> <span class="k">def</span> <span class="nf">is_reader_alive</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  92. <span class="sd">&quot;&quot;&quot;</span>
  93. <span class="sd"> Indicates whether or not the reader thread is alive.</span>
  94. <span class="sd"> :returns: Whether or not the reader thread is alive.</span>
  95. <span class="sd"> &quot;&quot;&quot;</span>
  96. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_read_thread</span><span class="o">.</span><span class="n">is_alive</span><span class="p">()</span>
  97. </div>
  98. <div class="viewcode-block" id="Device.stop_reader"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.Device.stop_reader">[docs]</a> <span class="k">def</span> <span class="nf">stop_reader</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  99. <span class="sd">&quot;&quot;&quot;</span>
  100. <span class="sd"> Stops the reader thread.</span>
  101. <span class="sd"> &quot;&quot;&quot;</span>
  102. <span class="bp">self</span><span class="o">.</span><span class="n">_read_thread</span><span class="o">.</span><span class="n">stop</span><span class="p">()</span>
  103. </div>
  104. <div class="viewcode-block" id="Device.close"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.Device.close">[docs]</a> <span class="k">def</span> <span class="nf">close</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  105. <span class="sd">&quot;&quot;&quot;</span>
  106. <span class="sd"> Closes the device.</span>
  107. <span class="sd"> &quot;&quot;&quot;</span>
  108. <span class="k">try</span><span class="p">:</span>
  109. <span class="bp">self</span><span class="o">.</span><span class="n">_running</span> <span class="o">=</span> <span class="bp">False</span>
  110. <span class="bp">self</span><span class="o">.</span><span class="n">_read_thread</span><span class="o">.</span><span class="n">stop</span><span class="p">()</span>
  111. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">close</span><span class="p">()</span>
  112. <span class="k">except</span><span class="p">:</span>
  113. <span class="k">pass</span>
  114. <span class="bp">self</span><span class="o">.</span><span class="n">on_close</span><span class="p">()</span>
  115. </div>
  116. <div class="viewcode-block" id="Device.ReadThread"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.Device.ReadThread">[docs]</a> <span class="k">class</span> <span class="nc">ReadThread</span><span class="p">(</span><span class="n">threading</span><span class="o">.</span><span class="n">Thread</span><span class="p">):</span>
  117. <span class="sd">&quot;&quot;&quot;</span>
  118. <span class="sd"> Reader thread which processes messages from the device.</span>
  119. <span class="sd"> &quot;&quot;&quot;</span>
  120. <span class="n">READ_TIMEOUT</span> <span class="o">=</span> <span class="mi">10</span>
  121. <span class="sd">&quot;&quot;&quot;Timeout for the reader thread.&quot;&quot;&quot;</span>
  122. <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">device</span><span class="p">):</span>
  123. <span class="sd">&quot;&quot;&quot;</span>
  124. <span class="sd"> Constructor</span>
  125. <span class="sd"> :param device: The device used by the reader thread.</span>
  126. <span class="sd"> :type device: devices.Device</span>
  127. <span class="sd"> &quot;&quot;&quot;</span>
  128. <span class="n">threading</span><span class="o">.</span><span class="n">Thread</span><span class="o">.</span><span class="n">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
  129. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span> <span class="o">=</span> <span class="n">device</span>
  130. <span class="bp">self</span><span class="o">.</span><span class="n">_running</span> <span class="o">=</span> <span class="bp">False</span>
  131. <div class="viewcode-block" id="Device.ReadThread.stop"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.Device.ReadThread.stop">[docs]</a> <span class="k">def</span> <span class="nf">stop</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  132. <span class="sd">&quot;&quot;&quot;</span>
  133. <span class="sd"> Stops the running thread.</span>
  134. <span class="sd"> &quot;&quot;&quot;</span>
  135. <span class="bp">self</span><span class="o">.</span><span class="n">_running</span> <span class="o">=</span> <span class="bp">False</span>
  136. </div>
  137. <div class="viewcode-block" id="Device.ReadThread.run"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.Device.ReadThread.run">[docs]</a> <span class="k">def</span> <span class="nf">run</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  138. <span class="sd">&quot;&quot;&quot;</span>
  139. <span class="sd"> The actual read process.</span>
  140. <span class="sd"> &quot;&quot;&quot;</span>
  141. <span class="bp">self</span><span class="o">.</span><span class="n">_running</span> <span class="o">=</span> <span class="bp">True</span>
  142. <span class="k">while</span> <span class="bp">self</span><span class="o">.</span><span class="n">_running</span><span class="p">:</span>
  143. <span class="k">try</span><span class="p">:</span>
  144. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">read_line</span><span class="p">(</span><span class="n">timeout</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">READ_TIMEOUT</span><span class="p">)</span>
  145. <span class="k">except</span> <span class="n">TimeoutError</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  146. <span class="k">pass</span>
  147. <span class="k">except</span> <span class="ne">Exception</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  148. <span class="bp">self</span><span class="o">.</span><span class="n">_running</span> <span class="o">=</span> <span class="bp">False</span>
  149. <span class="c">#raise err</span>
  150. <span class="n">time</span><span class="o">.</span><span class="n">sleep</span><span class="p">(</span><span class="mf">0.01</span><span class="p">)</span>
  151. </div></div></div>
  152. <div class="viewcode-block" id="USBDevice"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice">[docs]</a><span class="k">class</span> <span class="nc">USBDevice</span><span class="p">(</span><span class="n">Device</span><span class="p">):</span>
  153. <span class="sd">&quot;&quot;&quot;</span>
  154. <span class="sd"> AD2USB device exposed with PyFTDI&#39;s interface.</span>
  155. <span class="sd"> &quot;&quot;&quot;</span>
  156. <span class="c"># Constants</span>
  157. <span class="n">FTDI_VENDOR_ID</span> <span class="o">=</span> <span class="mh">0x0403</span>
  158. <span class="sd">&quot;&quot;&quot;Vendor ID used to recognize AD2USB devices.&quot;&quot;&quot;</span>
  159. <span class="n">FTDI_PRODUCT_ID</span> <span class="o">=</span> <span class="mh">0x6001</span>
  160. <span class="sd">&quot;&quot;&quot;Product ID used to recognize AD2USB devices.&quot;&quot;&quot;</span>
  161. <span class="n">BAUDRATE</span> <span class="o">=</span> <span class="mi">115200</span>
  162. <span class="sd">&quot;&quot;&quot;Default baudrate for AD2USB devices.&quot;&quot;&quot;</span>
  163. <span class="nd">@staticmethod</span>
  164. <div class="viewcode-block" id="USBDevice.find_all"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice.find_all">[docs]</a> <span class="k">def</span> <span class="nf">find_all</span><span class="p">():</span>
  165. <span class="sd">&quot;&quot;&quot;</span>
  166. <span class="sd"> Returns all FTDI devices matching our vendor and product IDs.</span>
  167. <span class="sd"> :returns: list of devices</span>
  168. <span class="sd"> :raises: CommError</span>
  169. <span class="sd"> &quot;&quot;&quot;</span>
  170. <span class="n">devices</span> <span class="o">=</span> <span class="p">[]</span>
  171. <span class="k">try</span><span class="p">:</span>
  172. <span class="n">devices</span> <span class="o">=</span> <span class="n">Ftdi</span><span class="o">.</span><span class="n">find_all</span><span class="p">([(</span><span class="n">USBDevice</span><span class="o">.</span><span class="n">FTDI_VENDOR_ID</span><span class="p">,</span> <span class="n">USBDevice</span><span class="o">.</span><span class="n">FTDI_PRODUCT_ID</span><span class="p">)],</span> <span class="n">nocache</span><span class="o">=</span><span class="bp">True</span><span class="p">)</span>
  173. <span class="k">except</span> <span class="p">(</span><span class="n">usb</span><span class="o">.</span><span class="n">core</span><span class="o">.</span><span class="n">USBError</span><span class="p">,</span> <span class="n">FtdiError</span><span class="p">),</span> <span class="n">err</span><span class="p">:</span>
  174. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error enumerating AD2USB devices: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  175. <span class="k">return</span> <span class="n">devices</span>
  176. </div>
  177. <span class="nd">@property</span>
  178. <span class="k">def</span> <span class="nf">interface</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  179. <span class="sd">&quot;&quot;&quot;</span>
  180. <span class="sd"> Retrieves the interface used to connect to the device.</span>
  181. <span class="sd"> :returns: the interface used to connect to the device.</span>
  182. <span class="sd"> &quot;&quot;&quot;</span>
  183. <span class="k">return</span> <span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_serial_number</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_endpoint</span><span class="p">)</span>
  184. <span class="nd">@interface.setter</span>
  185. <div class="viewcode-block" id="USBDevice.interface"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice.interface">[docs]</a> <span class="k">def</span> <span class="nf">interface</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  186. <span class="sd">&quot;&quot;&quot;</span>
  187. <span class="sd"> Sets the interface used to connect to the device.</span>
  188. <span class="sd"> :param value: Tuple containing the serial number and endpoint number to use.</span>
  189. <span class="sd"> :type value: tuple</span>
  190. <span class="sd"> &quot;&quot;&quot;</span>
  191. <span class="bp">self</span><span class="o">.</span><span class="n">_serial_number</span> <span class="o">=</span> <span class="n">value</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
  192. <span class="bp">self</span><span class="o">.</span><span class="n">_endpoint</span> <span class="o">=</span> <span class="n">value</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span>
  193. </div>
  194. <span class="nd">@property</span>
  195. <span class="k">def</span> <span class="nf">serial_number</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  196. <span class="sd">&quot;&quot;&quot;</span>
  197. <span class="sd"> Retrieves the serial number of the device.</span>
  198. <span class="sd"> :returns: The serial number of the device.</span>
  199. <span class="sd"> &quot;&quot;&quot;</span>
  200. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_serial_number</span>
  201. <span class="nd">@serial_number.setter</span>
  202. <div class="viewcode-block" id="USBDevice.serial_number"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice.serial_number">[docs]</a> <span class="k">def</span> <span class="nf">serial_number</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  203. <span class="sd">&quot;&quot;&quot;</span>
  204. <span class="sd"> Sets the serial number of the device.</span>
  205. <span class="sd"> :param value: The serial number of the device.</span>
  206. <span class="sd"> :type value: string</span>
  207. <span class="sd"> &quot;&quot;&quot;</span>
  208. <span class="bp">self</span><span class="o">.</span><span class="n">_serial_number</span> <span class="o">=</span> <span class="n">value</span>
  209. </div>
  210. <span class="nd">@property</span>
  211. <span class="k">def</span> <span class="nf">description</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  212. <span class="sd">&quot;&quot;&quot;</span>
  213. <span class="sd"> Retrieves the description of the device.</span>
  214. <span class="sd"> :returns: The description of the device.</span>
  215. <span class="sd"> &quot;&quot;&quot;</span>
  216. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_description</span>
  217. <span class="nd">@description.setter</span>
  218. <div class="viewcode-block" id="USBDevice.description"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice.description">[docs]</a> <span class="k">def</span> <span class="nf">description</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  219. <span class="sd">&quot;&quot;&quot;</span>
  220. <span class="sd"> Sets the description of the device.</span>
  221. <span class="sd"> :param value: The description of the device.</span>
  222. <span class="sd"> :type value: string</span>
  223. <span class="sd"> &quot;&quot;&quot;</span>
  224. <span class="bp">self</span><span class="o">.</span><span class="n">_description</span> <span class="o">=</span> <span class="n">value</span>
  225. </div>
  226. <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">interface</span><span class="o">=</span><span class="p">(</span><span class="bp">None</span><span class="p">,</span> <span class="mi">0</span><span class="p">)):</span>
  227. <span class="sd">&quot;&quot;&quot;</span>
  228. <span class="sd"> Constructor</span>
  229. <span class="sd"> :param interface: Tuple containing the device number and endpoint number to use.</span>
  230. <span class="sd"> :type interface: tuple</span>
  231. <span class="sd"> &quot;&quot;&quot;</span>
  232. <span class="n">Device</span><span class="o">.</span><span class="n">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
  233. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span> <span class="o">=</span> <span class="n">Ftdi</span><span class="p">()</span>
  234. <span class="bp">self</span><span class="o">.</span><span class="n">_serial_number</span> <span class="o">=</span> <span class="n">interface</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
  235. <span class="bp">self</span><span class="o">.</span><span class="n">_endpoint</span> <span class="o">=</span> <span class="n">interface</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span>
  236. <span class="bp">self</span><span class="o">.</span><span class="n">_vendor_id</span> <span class="o">=</span> <span class="n">USBDevice</span><span class="o">.</span><span class="n">FTDI_VENDOR_ID</span>
  237. <span class="bp">self</span><span class="o">.</span><span class="n">_product_id</span> <span class="o">=</span> <span class="n">USBDevice</span><span class="o">.</span><span class="n">FTDI_PRODUCT_ID</span>
  238. <span class="bp">self</span><span class="o">.</span><span class="n">_device_number</span> <span class="o">=</span> <span class="bp">None</span>
  239. <span class="bp">self</span><span class="o">.</span><span class="n">_description</span> <span class="o">=</span> <span class="bp">None</span>
  240. <div class="viewcode-block" id="USBDevice.open"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice.open">[docs]</a> <span class="k">def</span> <span class="nf">open</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">baudrate</span><span class="o">=</span><span class="n">BAUDRATE</span><span class="p">,</span> <span class="n">no_reader_thread</span><span class="o">=</span><span class="bp">False</span><span class="p">):</span>
  241. <span class="sd">&quot;&quot;&quot;</span>
  242. <span class="sd"> Opens the device.</span>
  243. <span class="sd"> :param baudrate: The baudrate to use.</span>
  244. <span class="sd"> :type baudrate: int</span>
  245. <span class="sd"> :param no_reader_thread: Whether or not to automatically start the reader thread.</span>
  246. <span class="sd"> :type no_reader_thread: bool</span>
  247. <span class="sd"> :raises: NoDeviceError</span>
  248. <span class="sd"> &quot;&quot;&quot;</span>
  249. <span class="c"># Set up defaults</span>
  250. <span class="k">if</span> <span class="n">baudrate</span> <span class="ow">is</span> <span class="bp">None</span><span class="p">:</span>
  251. <span class="n">baudrate</span> <span class="o">=</span> <span class="n">USBDevice</span><span class="o">.</span><span class="n">BAUDRATE</span>
  252. <span class="c"># Open the device and start up the thread.</span>
  253. <span class="k">try</span><span class="p">:</span>
  254. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">open</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_vendor_id</span><span class="p">,</span>
  255. <span class="bp">self</span><span class="o">.</span><span class="n">_product_id</span><span class="p">,</span>
  256. <span class="bp">self</span><span class="o">.</span><span class="n">_endpoint</span><span class="p">,</span>
  257. <span class="bp">self</span><span class="o">.</span><span class="n">_device_number</span><span class="p">,</span>
  258. <span class="bp">self</span><span class="o">.</span><span class="n">_serial_number</span><span class="p">,</span>
  259. <span class="bp">self</span><span class="o">.</span><span class="n">_description</span><span class="p">)</span>
  260. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">set_baudrate</span><span class="p">(</span><span class="n">baudrate</span><span class="p">)</span>
  261. <span class="bp">self</span><span class="o">.</span><span class="n">_id</span> <span class="o">=</span> <span class="s">&#39;USB {0}:{1}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">usb_dev</span><span class="o">.</span><span class="n">bus</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">usb_dev</span><span class="o">.</span><span class="n">address</span><span class="p">)</span>
  262. <span class="k">except</span> <span class="p">(</span><span class="n">usb</span><span class="o">.</span><span class="n">core</span><span class="o">.</span><span class="n">USBError</span><span class="p">,</span> <span class="n">FtdiError</span><span class="p">),</span> <span class="n">err</span><span class="p">:</span>
  263. <span class="k">raise</span> <span class="n">NoDeviceError</span><span class="p">(</span><span class="s">&#39;Error opening device: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  264. <span class="k">else</span><span class="p">:</span>
  265. <span class="bp">self</span><span class="o">.</span><span class="n">_running</span> <span class="o">=</span> <span class="bp">True</span>
  266. <span class="k">if</span> <span class="ow">not</span> <span class="n">no_reader_thread</span><span class="p">:</span>
  267. <span class="bp">self</span><span class="o">.</span><span class="n">_read_thread</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>
  268. <span class="bp">self</span><span class="o">.</span><span class="n">on_open</span><span class="p">()</span>
  269. </div>
  270. <div class="viewcode-block" id="USBDevice.close"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice.close">[docs]</a> <span class="k">def</span> <span class="nf">close</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  271. <span class="sd">&quot;&quot;&quot;</span>
  272. <span class="sd"> Closes the device.</span>
  273. <span class="sd"> &quot;&quot;&quot;</span>
  274. <span class="k">try</span><span class="p">:</span>
  275. <span class="c"># HACK: Probably should fork pyftdi and make this call in .close().</span>
  276. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">usb_dev</span><span class="o">.</span><span class="n">attach_kernel_driver</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_device_number</span><span class="p">)</span>
  277. <span class="n">Device</span><span class="o">.</span><span class="n">close</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
  278. <span class="k">except</span><span class="p">:</span>
  279. <span class="k">pass</span>
  280. </div>
  281. <div class="viewcode-block" id="USBDevice.write"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice.write">[docs]</a> <span class="k">def</span> <span class="nf">write</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">data</span><span class="p">):</span>
  282. <span class="sd">&quot;&quot;&quot;</span>
  283. <span class="sd"> Writes data to the device.</span>
  284. <span class="sd"> :param data: Data to write</span>
  285. <span class="sd"> :type data: str</span>
  286. <span class="sd"> :raises: CommError</span>
  287. <span class="sd"> &quot;&quot;&quot;</span>
  288. <span class="k">try</span><span class="p">:</span>
  289. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">write_data</span><span class="p">(</span><span class="n">data</span><span class="p">)</span>
  290. <span class="bp">self</span><span class="o">.</span><span class="n">on_write</span><span class="p">(</span><span class="n">data</span><span class="p">)</span>
  291. <span class="k">except</span> <span class="n">FtdiError</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  292. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error writing to device: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  293. </div>
  294. <div class="viewcode-block" id="USBDevice.read"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice.read">[docs]</a> <span class="k">def</span> <span class="nf">read</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  295. <span class="sd">&quot;&quot;&quot;</span>
  296. <span class="sd"> Reads a single character from the device.</span>
  297. <span class="sd"> :returns: The character read from the device.</span>
  298. <span class="sd"> :raises: CommError</span>
  299. <span class="sd"> &quot;&quot;&quot;</span>
  300. <span class="n">ret</span> <span class="o">=</span> <span class="bp">None</span>
  301. <span class="k">try</span><span class="p">:</span>
  302. <span class="n">ret</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">read_data</span><span class="p">(</span><span class="mi">1</span><span class="p">)</span>
  303. <span class="k">except</span> <span class="p">(</span><span class="n">usb</span><span class="o">.</span><span class="n">core</span><span class="o">.</span><span class="n">USBError</span><span class="p">,</span> <span class="n">FtdiError</span><span class="p">),</span> <span class="n">err</span><span class="p">:</span>
  304. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error reading from device: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  305. <span class="k">return</span> <span class="n">ret</span>
  306. </div>
  307. <div class="viewcode-block" id="USBDevice.read_line"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.USBDevice.read_line">[docs]</a> <span class="k">def</span> <span class="nf">read_line</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">timeout</span><span class="o">=</span><span class="mf">0.0</span><span class="p">,</span> <span class="n">purge_buffer</span><span class="o">=</span><span class="bp">False</span><span class="p">):</span>
  308. <span class="sd">&quot;&quot;&quot;</span>
  309. <span class="sd"> Reads a line from the device.</span>
  310. <span class="sd"> :param timeout: Read timeout</span>
  311. <span class="sd"> :type timeout: float</span>
  312. <span class="sd"> :param purge_buffer: Indicates whether to purge the buffer prior to reading.</span>
  313. <span class="sd"> :type purge_buffer: bool</span>
  314. <span class="sd"> :returns: The line that was read.</span>
  315. <span class="sd"> :raises: CommError, TimeoutError</span>
  316. <span class="sd"> &quot;&quot;&quot;</span>
  317. <span class="k">if</span> <span class="n">purge_buffer</span><span class="p">:</span>
  318. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>
  319. <span class="k">def</span> <span class="nf">timeout_event</span><span class="p">():</span>
  320. <span class="n">timeout_event</span><span class="o">.</span><span class="n">reading</span> <span class="o">=</span> <span class="bp">False</span>
  321. <span class="n">timeout_event</span><span class="o">.</span><span class="n">reading</span> <span class="o">=</span> <span class="bp">True</span>
  322. <span class="n">got_line</span> <span class="o">=</span> <span class="bp">False</span>
  323. <span class="n">ret</span> <span class="o">=</span> <span class="bp">None</span>
  324. <span class="n">timer</span> <span class="o">=</span> <span class="bp">None</span>
  325. <span class="k">if</span> <span class="n">timeout</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">:</span>
  326. <span class="n">timer</span> <span class="o">=</span> <span class="n">threading</span><span class="o">.</span><span class="n">Timer</span><span class="p">(</span><span class="n">timeout</span><span class="p">,</span> <span class="n">timeout_event</span><span class="p">)</span>
  327. <span class="n">timer</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>
  328. <span class="k">try</span><span class="p">:</span>
  329. <span class="k">while</span> <span class="n">timeout_event</span><span class="o">.</span><span class="n">reading</span><span class="p">:</span>
  330. <span class="n">buf</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">read_data</span><span class="p">(</span><span class="mi">1</span><span class="p">)</span>
  331. <span class="k">if</span> <span class="n">buf</span> <span class="o">!=</span> <span class="s">&#39;&#39;</span><span class="p">:</span>
  332. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">+=</span> <span class="n">buf</span>
  333. <span class="k">if</span> <span class="n">buf</span> <span class="o">==</span> <span class="s">&quot;</span><span class="se">\n</span><span class="s">&quot;</span><span class="p">:</span>
  334. <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">)</span> <span class="o">&gt;</span> <span class="mi">1</span><span class="p">:</span>
  335. <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">[</span><span class="o">-</span><span class="mi">2</span><span class="p">]</span> <span class="o">==</span> <span class="s">&quot;</span><span class="se">\r</span><span class="s">&quot;</span><span class="p">:</span>
  336. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">[:</span><span class="o">-</span><span class="mi">2</span><span class="p">]</span>
  337. <span class="c"># ignore if we just got \r\n with nothing else in the buffer.</span>
  338. <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">)</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">:</span>
  339. <span class="n">got_line</span> <span class="o">=</span> <span class="bp">True</span>
  340. <span class="k">break</span>
  341. <span class="k">else</span><span class="p">:</span>
  342. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span>
  343. <span class="k">except</span> <span class="p">(</span><span class="n">usb</span><span class="o">.</span><span class="n">core</span><span class="o">.</span><span class="n">USBError</span><span class="p">,</span> <span class="n">FtdiError</span><span class="p">),</span> <span class="n">err</span><span class="p">:</span>
  344. <span class="k">if</span> <span class="n">timer</span><span class="p">:</span>
  345. <span class="n">timer</span><span class="o">.</span><span class="n">cancel</span><span class="p">()</span>
  346. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error reading from device: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  347. <span class="k">else</span><span class="p">:</span>
  348. <span class="k">if</span> <span class="n">got_line</span><span class="p">:</span>
  349. <span class="n">ret</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span>
  350. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>
  351. <span class="bp">self</span><span class="o">.</span><span class="n">on_read</span><span class="p">(</span><span class="n">ret</span><span class="p">)</span>
  352. <span class="k">if</span> <span class="n">timer</span><span class="p">:</span>
  353. <span class="k">if</span> <span class="n">timer</span><span class="o">.</span><span class="n">is_alive</span><span class="p">():</span>
  354. <span class="n">timer</span><span class="o">.</span><span class="n">cancel</span><span class="p">()</span>
  355. <span class="k">else</span><span class="p">:</span>
  356. <span class="k">raise</span> <span class="n">TimeoutError</span><span class="p">(</span><span class="s">&#39;Timeout while waiting for line terminator.&#39;</span><span class="p">)</span>
  357. <span class="k">return</span> <span class="n">ret</span>
  358. </div></div>
  359. <div class="viewcode-block" id="SerialDevice"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SerialDevice">[docs]</a><span class="k">class</span> <span class="nc">SerialDevice</span><span class="p">(</span><span class="n">Device</span><span class="p">):</span>
  360. <span class="sd">&quot;&quot;&quot;</span>
  361. <span class="sd"> AD2USB or AD2SERIAL device exposed with the pyserial interface.</span>
  362. <span class="sd"> &quot;&quot;&quot;</span>
  363. <span class="c"># Constants</span>
  364. <span class="n">BAUDRATE</span> <span class="o">=</span> <span class="mi">19200</span>
  365. <span class="sd">&quot;&quot;&quot;Default baudrate for Serial devices.&quot;&quot;&quot;</span>
  366. <span class="nd">@staticmethod</span>
  367. <div class="viewcode-block" id="SerialDevice.find_all"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SerialDevice.find_all">[docs]</a> <span class="k">def</span> <span class="nf">find_all</span><span class="p">(</span><span class="n">pattern</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
  368. <span class="sd">&quot;&quot;&quot;</span>
  369. <span class="sd"> Returns all serial ports present.</span>
  370. <span class="sd"> :param pattern: Pattern to search for when retrieving serial ports.</span>
  371. <span class="sd"> :type pattern: str</span>
  372. <span class="sd"> :returns: list of devices</span>
  373. <span class="sd"> :raises: CommError</span>
  374. <span class="sd"> &quot;&quot;&quot;</span>
  375. <span class="n">devices</span> <span class="o">=</span> <span class="p">[]</span>
  376. <span class="k">try</span><span class="p">:</span>
  377. <span class="k">if</span> <span class="n">pattern</span><span class="p">:</span>
  378. <span class="n">devices</span> <span class="o">=</span> <span class="n">serial</span><span class="o">.</span><span class="n">tools</span><span class="o">.</span><span class="n">list_ports</span><span class="o">.</span><span class="n">grep</span><span class="p">(</span><span class="n">pattern</span><span class="p">)</span>
  379. <span class="k">else</span><span class="p">:</span>
  380. <span class="n">devices</span> <span class="o">=</span> <span class="n">serial</span><span class="o">.</span><span class="n">tools</span><span class="o">.</span><span class="n">list_ports</span><span class="o">.</span><span class="n">comports</span><span class="p">()</span>
  381. <span class="k">except</span> <span class="n">SerialException</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  382. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error enumerating serial devices: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  383. <span class="k">return</span> <span class="n">devices</span>
  384. </div>
  385. <span class="nd">@property</span>
  386. <span class="k">def</span> <span class="nf">interface</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  387. <span class="sd">&quot;&quot;&quot;</span>
  388. <span class="sd"> Retrieves the interface used to connect to the device.</span>
  389. <span class="sd"> :returns: the interface used to connect to the device.</span>
  390. <span class="sd"> &quot;&quot;&quot;</span>
  391. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_port</span>
  392. <span class="nd">@interface.setter</span>
  393. <div class="viewcode-block" id="SerialDevice.interface"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SerialDevice.interface">[docs]</a> <span class="k">def</span> <span class="nf">interface</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  394. <span class="sd">&quot;&quot;&quot;</span>
  395. <span class="sd"> Sets the interface used to connect to the device.</span>
  396. <span class="sd"> :param value: The name of the serial device.</span>
  397. <span class="sd"> :type value: string</span>
  398. <span class="sd"> &quot;&quot;&quot;</span>
  399. <span class="bp">self</span><span class="o">.</span><span class="n">_port</span> <span class="o">=</span> <span class="n">value</span>
  400. </div>
  401. <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">interface</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
  402. <span class="sd">&quot;&quot;&quot;</span>
  403. <span class="sd"> Constructor</span>
  404. <span class="sd"> :param interface: The device to open.</span>
  405. <span class="sd"> :type interface: str</span>
  406. <span class="sd"> &quot;&quot;&quot;</span>
  407. <span class="n">Device</span><span class="o">.</span><span class="n">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
  408. <span class="bp">self</span><span class="o">.</span><span class="n">_port</span> <span class="o">=</span> <span class="n">interface</span>
  409. <span class="bp">self</span><span class="o">.</span><span class="n">_id</span> <span class="o">=</span> <span class="n">interface</span>
  410. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span> <span class="o">=</span> <span class="n">serial</span><span class="o">.</span><span class="n">Serial</span><span class="p">(</span><span class="n">timeout</span><span class="o">=</span><span class="mi">0</span><span class="p">,</span> <span class="n">writeTimeout</span><span class="o">=</span><span class="mi">0</span><span class="p">)</span> <span class="c"># Timeout = non-blocking to match pyftdi.</span>
  411. <div class="viewcode-block" id="SerialDevice.open"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SerialDevice.open">[docs]</a> <span class="k">def</span> <span class="nf">open</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">baudrate</span><span class="o">=</span><span class="n">BAUDRATE</span><span class="p">,</span> <span class="n">no_reader_thread</span><span class="o">=</span><span class="bp">False</span><span class="p">):</span>
  412. <span class="sd">&quot;&quot;&quot;</span>
  413. <span class="sd"> Opens the device.</span>
  414. <span class="sd"> :param baudrate: The baudrate to use with the device.</span>
  415. <span class="sd"> :type baudrate: int</span>
  416. <span class="sd"> :param no_reader_thread: Whether or not to automatically start the reader thread.</span>
  417. <span class="sd"> :type no_reader_thread: bool</span>
  418. <span class="sd"> :raises: NoDeviceError</span>
  419. <span class="sd"> &quot;&quot;&quot;</span>
  420. <span class="c"># Set up the defaults</span>
  421. <span class="k">if</span> <span class="n">baudrate</span> <span class="ow">is</span> <span class="bp">None</span><span class="p">:</span>
  422. <span class="n">baudrate</span> <span class="o">=</span> <span class="n">SerialDevice</span><span class="o">.</span><span class="n">BAUDRATE</span>
  423. <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_port</span> <span class="ow">is</span> <span class="bp">None</span><span class="p">:</span>
  424. <span class="k">raise</span> <span class="n">NoDeviceError</span><span class="p">(</span><span class="s">&#39;No device interface specified.&#39;</span><span class="p">)</span>
  425. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">port</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_port</span>
  426. <span class="c"># Open the device and start up the reader thread.</span>
  427. <span class="k">try</span><span class="p">:</span>
  428. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">open</span><span class="p">()</span>
  429. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">baudrate</span> <span class="o">=</span> <span class="n">baudrate</span> <span class="c"># NOTE: Setting the baudrate before opening the</span>
  430. <span class="c"># port caused issues with Moschip 7840/7820</span>
  431. <span class="c"># USB Serial Driver converter. (mos7840)</span>
  432. <span class="c">#</span>
  433. <span class="c"># Moving it to this point seems to resolve</span>
  434. <span class="c"># all issues with it.</span>
  435. <span class="k">except</span> <span class="p">(</span><span class="n">serial</span><span class="o">.</span><span class="n">SerialException</span><span class="p">,</span> <span class="ne">ValueError</span><span class="p">),</span> <span class="n">err</span><span class="p">:</span>
  436. <span class="k">raise</span> <span class="n">NoDeviceError</span><span class="p">(</span><span class="s">&#39;Error opening device on port {0}.&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_port</span><span class="p">),</span> <span class="n">err</span><span class="p">)</span>
  437. <span class="k">else</span><span class="p">:</span>
  438. <span class="bp">self</span><span class="o">.</span><span class="n">_running</span> <span class="o">=</span> <span class="bp">True</span>
  439. <span class="bp">self</span><span class="o">.</span><span class="n">on_open</span><span class="p">()</span>
  440. <span class="k">if</span> <span class="ow">not</span> <span class="n">no_reader_thread</span><span class="p">:</span>
  441. <span class="bp">self</span><span class="o">.</span><span class="n">_read_thread</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>
  442. </div>
  443. <div class="viewcode-block" id="SerialDevice.close"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SerialDevice.close">[docs]</a> <span class="k">def</span> <span class="nf">close</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  444. <span class="sd">&quot;&quot;&quot;</span>
  445. <span class="sd"> Closes the device.</span>
  446. <span class="sd"> &quot;&quot;&quot;</span>
  447. <span class="k">try</span><span class="p">:</span>
  448. <span class="n">Device</span><span class="o">.</span><span class="n">close</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
  449. <span class="k">except</span><span class="p">:</span>
  450. <span class="k">pass</span>
  451. </div>
  452. <div class="viewcode-block" id="SerialDevice.write"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SerialDevice.write">[docs]</a> <span class="k">def</span> <span class="nf">write</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">data</span><span class="p">):</span>
  453. <span class="sd">&quot;&quot;&quot;</span>
  454. <span class="sd"> Writes data to the device.</span>
  455. <span class="sd"> :param data: The data to write.</span>
  456. <span class="sd"> :type data: str</span>
  457. <span class="sd"> :raises: CommError</span>
  458. <span class="sd"> &quot;&quot;&quot;</span>
  459. <span class="k">try</span><span class="p">:</span>
  460. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="n">data</span><span class="p">)</span>
  461. <span class="k">except</span> <span class="n">serial</span><span class="o">.</span><span class="n">SerialTimeoutException</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  462. <span class="k">pass</span>
  463. <span class="k">except</span> <span class="n">serial</span><span class="o">.</span><span class="n">SerialException</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  464. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error writing to device.&#39;</span><span class="p">,</span> <span class="n">err</span><span class="p">)</span>
  465. <span class="k">else</span><span class="p">:</span>
  466. <span class="bp">self</span><span class="o">.</span><span class="n">on_write</span><span class="p">(</span><span class="n">data</span><span class="p">)</span>
  467. </div>
  468. <div class="viewcode-block" id="SerialDevice.read"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SerialDevice.read">[docs]</a> <span class="k">def</span> <span class="nf">read</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  469. <span class="sd">&quot;&quot;&quot;</span>
  470. <span class="sd"> Reads a single character from the device.</span>
  471. <span class="sd"> :returns: The character read from the device.</span>
  472. <span class="sd"> :raises: CommError</span>
  473. <span class="sd"> &quot;&quot;&quot;</span>
  474. <span class="n">ret</span> <span class="o">=</span> <span class="bp">None</span>
  475. <span class="k">try</span><span class="p">:</span>
  476. <span class="n">ret</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">read</span><span class="p">(</span><span class="mi">1</span><span class="p">)</span>
  477. <span class="k">except</span> <span class="n">serial</span><span class="o">.</span><span class="n">SerialException</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  478. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error reading from device: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  479. <span class="k">return</span> <span class="n">ret</span>
  480. </div>
  481. <div class="viewcode-block" id="SerialDevice.read_line"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SerialDevice.read_line">[docs]</a> <span class="k">def</span> <span class="nf">read_line</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">timeout</span><span class="o">=</span><span class="mf">0.0</span><span class="p">,</span> <span class="n">purge_buffer</span><span class="o">=</span><span class="bp">False</span><span class="p">):</span>
  482. <span class="sd">&quot;&quot;&quot;</span>
  483. <span class="sd"> Reads a line from the device.</span>
  484. <span class="sd"> :param timeout: The read timeout.</span>
  485. <span class="sd"> :type timeout: float</span>
  486. <span class="sd"> :param purge_buffer: Indicates whether to purge the buffer prior to reading.</span>
  487. <span class="sd"> :type purge_buffer: bool</span>
  488. <span class="sd"> :returns: The line read.</span>
  489. <span class="sd"> :raises: CommError, TimeoutError</span>
  490. <span class="sd"> &quot;&quot;&quot;</span>
  491. <span class="k">def</span> <span class="nf">timeout_event</span><span class="p">():</span>
  492. <span class="n">timeout_event</span><span class="o">.</span><span class="n">reading</span> <span class="o">=</span> <span class="bp">False</span>
  493. <span class="n">timeout_event</span><span class="o">.</span><span class="n">reading</span> <span class="o">=</span> <span class="bp">True</span>
  494. <span class="n">got_line</span> <span class="o">=</span> <span class="bp">False</span>
  495. <span class="n">ret</span> <span class="o">=</span> <span class="bp">None</span>
  496. <span class="n">timer</span> <span class="o">=</span> <span class="bp">None</span>
  497. <span class="k">if</span> <span class="n">timeout</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">:</span>
  498. <span class="n">timer</span> <span class="o">=</span> <span class="n">threading</span><span class="o">.</span><span class="n">Timer</span><span class="p">(</span><span class="n">timeout</span><span class="p">,</span> <span class="n">timeout_event</span><span class="p">)</span>
  499. <span class="n">timer</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>
  500. <span class="k">try</span><span class="p">:</span>
  501. <span class="k">while</span> <span class="n">timeout_event</span><span class="o">.</span><span class="n">reading</span><span class="p">:</span>
  502. <span class="n">buf</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">read</span><span class="p">(</span><span class="mi">1</span><span class="p">)</span>
  503. <span class="k">if</span> <span class="n">buf</span> <span class="o">!=</span> <span class="s">&#39;&#39;</span> <span class="ow">and</span> <span class="n">buf</span> <span class="o">!=</span> <span class="s">&quot;</span><span class="se">\xff</span><span class="s">&quot;</span><span class="p">:</span> <span class="c"># AD2SERIAL specifically apparently sends down \xFF on boot.</span>
  504. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">+=</span> <span class="n">buf</span>
  505. <span class="k">if</span> <span class="n">buf</span> <span class="o">==</span> <span class="s">&quot;</span><span class="se">\n</span><span class="s">&quot;</span><span class="p">:</span>
  506. <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">)</span> <span class="o">&gt;</span> <span class="mi">1</span><span class="p">:</span>
  507. <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">[</span><span class="o">-</span><span class="mi">2</span><span class="p">]</span> <span class="o">==</span> <span class="s">&quot;</span><span class="se">\r</span><span class="s">&quot;</span><span class="p">:</span>
  508. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">[:</span><span class="o">-</span><span class="mi">2</span><span class="p">]</span>
  509. <span class="c"># ignore if we just got \r\n with nothing else in the buffer.</span>
  510. <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">)</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">:</span>
  511. <span class="n">got_line</span> <span class="o">=</span> <span class="bp">True</span>
  512. <span class="k">break</span>
  513. <span class="k">else</span><span class="p">:</span>
  514. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span>
  515. <span class="k">except</span> <span class="p">(</span><span class="ne">OSError</span><span class="p">,</span> <span class="n">serial</span><span class="o">.</span><span class="n">SerialException</span><span class="p">),</span> <span class="n">err</span><span class="p">:</span>
  516. <span class="k">if</span> <span class="n">timer</span><span class="p">:</span>
  517. <span class="n">timer</span><span class="o">.</span><span class="n">cancel</span><span class="p">()</span>
  518. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error reading from device: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  519. <span class="k">else</span><span class="p">:</span>
  520. <span class="k">if</span> <span class="n">got_line</span><span class="p">:</span>
  521. <span class="n">ret</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span>
  522. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>
  523. <span class="bp">self</span><span class="o">.</span><span class="n">on_read</span><span class="p">(</span><span class="n">ret</span><span class="p">)</span>
  524. <span class="k">if</span> <span class="n">timer</span><span class="p">:</span>
  525. <span class="k">if</span> <span class="n">timer</span><span class="o">.</span><span class="n">is_alive</span><span class="p">():</span>
  526. <span class="n">timer</span><span class="o">.</span><span class="n">cancel</span><span class="p">()</span>
  527. <span class="k">else</span><span class="p">:</span>
  528. <span class="k">raise</span> <span class="n">TimeoutError</span><span class="p">(</span><span class="s">&#39;Timeout while waiting for line terminator.&#39;</span><span class="p">)</span>
  529. <span class="k">return</span> <span class="n">ret</span>
  530. </div></div>
  531. <div class="viewcode-block" id="SocketDevice"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice">[docs]</a><span class="k">class</span> <span class="nc">SocketDevice</span><span class="p">(</span><span class="n">Device</span><span class="p">):</span>
  532. <span class="sd">&quot;&quot;&quot;</span>
  533. <span class="sd"> Device that supports communication with an AD2 that is exposed via ser2sock or another</span>
  534. <span class="sd"> Serial to IP interface.</span>
  535. <span class="sd"> &quot;&quot;&quot;</span>
  536. <span class="nd">@property</span>
  537. <span class="k">def</span> <span class="nf">interface</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  538. <span class="sd">&quot;&quot;&quot;</span>
  539. <span class="sd"> Retrieves the interface used to connect to the device.</span>
  540. <span class="sd"> :returns: the interface used to connect to the device.</span>
  541. <span class="sd"> &quot;&quot;&quot;</span>
  542. <span class="k">return</span> <span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_host</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_port</span><span class="p">)</span>
  543. <span class="nd">@interface.setter</span>
  544. <div class="viewcode-block" id="SocketDevice.interface"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.interface">[docs]</a> <span class="k">def</span> <span class="nf">interface</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  545. <span class="sd">&quot;&quot;&quot;</span>
  546. <span class="sd"> Sets the interface used to connect to the device.</span>
  547. <span class="sd"> :param value: Tuple containing the device number and endpoint number to use.</span>
  548. <span class="sd"> :type value: tuple</span>
  549. <span class="sd"> &quot;&quot;&quot;</span>
  550. <span class="bp">self</span><span class="o">.</span><span class="n">_host</span> <span class="o">=</span> <span class="n">value</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
  551. <span class="bp">self</span><span class="o">.</span><span class="n">_port</span> <span class="o">=</span> <span class="n">value</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span>
  552. </div>
  553. <span class="nd">@property</span>
  554. <span class="k">def</span> <span class="nf">ssl</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  555. <span class="sd">&quot;&quot;&quot;</span>
  556. <span class="sd"> Retrieves whether or not the device is using SSL.</span>
  557. <span class="sd"> :returns: Whether or not the device is using SSL.</span>
  558. <span class="sd"> &quot;&quot;&quot;</span>
  559. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_use_ssl</span>
  560. <span class="nd">@ssl.setter</span>
  561. <div class="viewcode-block" id="SocketDevice.ssl"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.ssl">[docs]</a> <span class="k">def</span> <span class="nf">ssl</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  562. <span class="sd">&quot;&quot;&quot;</span>
  563. <span class="sd"> Sets whether or not SSL communication is in use.</span>
  564. <span class="sd"> :param value: Whether or not SSL communication is in use.</span>
  565. <span class="sd"> :type value: bool</span>
  566. <span class="sd"> &quot;&quot;&quot;</span>
  567. <span class="bp">self</span><span class="o">.</span><span class="n">_use_ssl</span> <span class="o">=</span> <span class="n">value</span>
  568. </div>
  569. <span class="nd">@property</span>
  570. <span class="k">def</span> <span class="nf">ssl_certificate</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  571. <span class="sd">&quot;&quot;&quot;</span>
  572. <span class="sd"> Retrieves the SSL client certificate path used for authentication.</span>
  573. <span class="sd"> :returns: The certificate path</span>
  574. <span class="sd"> &quot;&quot;&quot;</span>
  575. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_ssl_certificate</span>
  576. <span class="nd">@ssl_certificate.setter</span>
  577. <div class="viewcode-block" id="SocketDevice.ssl_certificate"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.ssl_certificate">[docs]</a> <span class="k">def</span> <span class="nf">ssl_certificate</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  578. <span class="sd">&quot;&quot;&quot;</span>
  579. <span class="sd"> Sets the SSL client certificate to use for authentication.</span>
  580. <span class="sd"> :param value: The path to the SSL certificate.</span>
  581. <span class="sd"> :type value: str</span>
  582. <span class="sd"> &quot;&quot;&quot;</span>
  583. <span class="bp">self</span><span class="o">.</span><span class="n">_ssl_certificate</span> <span class="o">=</span> <span class="n">value</span>
  584. </div>
  585. <span class="nd">@property</span>
  586. <span class="k">def</span> <span class="nf">ssl_key</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  587. <span class="sd">&quot;&quot;&quot;</span>
  588. <span class="sd"> Retrieves the SSL client certificate key used for authentication.</span>
  589. <span class="sd"> :returns: The key path</span>
  590. <span class="sd"> &quot;&quot;&quot;</span>
  591. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_ssl_key</span>
  592. <span class="nd">@ssl_key.setter</span>
  593. <div class="viewcode-block" id="SocketDevice.ssl_key"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.ssl_key">[docs]</a> <span class="k">def</span> <span class="nf">ssl_key</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  594. <span class="sd">&quot;&quot;&quot;</span>
  595. <span class="sd"> Sets the SSL client certificate key to use for authentication.</span>
  596. <span class="sd"> :param value: The path to the SSL key.</span>
  597. <span class="sd"> :type value: str</span>
  598. <span class="sd"> &quot;&quot;&quot;</span>
  599. <span class="bp">self</span><span class="o">.</span><span class="n">_ssl_key</span> <span class="o">=</span> <span class="n">value</span>
  600. </div>
  601. <span class="nd">@property</span>
  602. <span class="k">def</span> <span class="nf">ssl_ca</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  603. <span class="sd">&quot;&quot;&quot;</span>
  604. <span class="sd"> Retrieves the SSL Certificate Authority certificate used for authentication.</span>
  605. <span class="sd"> :returns: The CA path</span>
  606. <span class="sd"> &quot;&quot;&quot;</span>
  607. <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_ssl_ca</span>
  608. <span class="nd">@ssl_ca.setter</span>
  609. <div class="viewcode-block" id="SocketDevice.ssl_ca"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.ssl_ca">[docs]</a> <span class="k">def</span> <span class="nf">ssl_ca</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
  610. <span class="sd">&quot;&quot;&quot;</span>
  611. <span class="sd"> Sets the SSL Certificate Authority certificate used for authentication.</span>
  612. <span class="sd"> :param value: The path to the SSL CA certificate.</span>
  613. <span class="sd"> :type value: str</span>
  614. <span class="sd"> &quot;&quot;&quot;</span>
  615. <span class="bp">self</span><span class="o">.</span><span class="n">_ssl_ca</span> <span class="o">=</span> <span class="n">value</span>
  616. </div>
  617. <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">interface</span><span class="o">=</span><span class="p">(</span><span class="s">&quot;localhost&quot;</span><span class="p">,</span> <span class="mi">10000</span><span class="p">)):</span>
  618. <span class="sd">&quot;&quot;&quot;</span>
  619. <span class="sd"> Constructor</span>
  620. <span class="sd"> :param interface: Tuple containing the hostname and port of our target.</span>
  621. <span class="sd"> :type interface: tuple</span>
  622. <span class="sd"> &quot;&quot;&quot;</span>
  623. <span class="n">Device</span><span class="o">.</span><span class="n">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
  624. <span class="bp">self</span><span class="o">.</span><span class="n">_host</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_port</span> <span class="o">=</span> <span class="n">interface</span>
  625. <span class="bp">self</span><span class="o">.</span><span class="n">_use_ssl</span> <span class="o">=</span> <span class="bp">False</span>
  626. <span class="bp">self</span><span class="o">.</span><span class="n">_ssl_certificate</span> <span class="o">=</span> <span class="bp">None</span>
  627. <span class="bp">self</span><span class="o">.</span><span class="n">_ssl_key</span> <span class="o">=</span> <span class="bp">None</span>
  628. <span class="bp">self</span><span class="o">.</span><span class="n">_ssl_ca</span> <span class="o">=</span> <span class="bp">None</span>
  629. <div class="viewcode-block" id="SocketDevice.open"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.open">[docs]</a> <span class="k">def</span> <span class="nf">open</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">baudrate</span><span class="o">=</span><span class="bp">None</span><span class="p">,</span> <span class="n">no_reader_thread</span><span class="o">=</span><span class="bp">False</span><span class="p">):</span>
  630. <span class="sd">&quot;&quot;&quot;</span>
  631. <span class="sd"> Opens the device.</span>
  632. <span class="sd"> :param baudrate: The baudrate to use</span>
  633. <span class="sd"> :type baudrate: int</span>
  634. <span class="sd"> :param no_reader_thread: Whether or not to automatically open the reader thread.</span>
  635. <span class="sd"> :type no_reader_thread: bool</span>
  636. <span class="sd"> :raises: NoDeviceError, CommError</span>
  637. <span class="sd"> &quot;&quot;&quot;</span>
  638. <span class="k">try</span><span class="p">:</span>
  639. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span> <span class="o">=</span> <span class="n">socket</span><span class="o">.</span><span class="n">socket</span><span class="p">(</span><span class="n">socket</span><span class="o">.</span><span class="n">AF_INET</span><span class="p">,</span> <span class="n">socket</span><span class="o">.</span><span class="n">SOCK_STREAM</span><span class="p">)</span>
  640. <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_use_ssl</span><span class="p">:</span>
  641. <span class="bp">self</span><span class="o">.</span><span class="n">_init_ssl</span><span class="p">()</span>
  642. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">connect</span><span class="p">((</span><span class="bp">self</span><span class="o">.</span><span class="n">_host</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_port</span><span class="p">))</span>
  643. <span class="bp">self</span><span class="o">.</span><span class="n">_id</span> <span class="o">=</span> <span class="s">&#39;{0}:{1}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_host</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_port</span><span class="p">)</span>
  644. <span class="k">except</span> <span class="n">socket</span><span class="o">.</span><span class="n">error</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  645. <span class="k">raise</span> <span class="n">NoDeviceError</span><span class="p">(</span><span class="s">&#39;Error opening device at {0}:{1}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_host</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_port</span><span class="p">),</span> <span class="n">err</span><span class="p">)</span>
  646. <span class="k">else</span><span class="p">:</span>
  647. <span class="bp">self</span><span class="o">.</span><span class="n">_running</span> <span class="o">=</span> <span class="bp">True</span>
  648. <span class="bp">self</span><span class="o">.</span><span class="n">on_open</span><span class="p">()</span>
  649. <span class="k">if</span> <span class="ow">not</span> <span class="n">no_reader_thread</span><span class="p">:</span>
  650. <span class="bp">self</span><span class="o">.</span><span class="n">_read_thread</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>
  651. </div>
  652. <div class="viewcode-block" id="SocketDevice.close"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.close">[docs]</a> <span class="k">def</span> <span class="nf">close</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  653. <span class="sd">&quot;&quot;&quot;</span>
  654. <span class="sd"> Closes the device.</span>
  655. <span class="sd"> &quot;&quot;&quot;</span>
  656. <span class="k">try</span><span class="p">:</span>
  657. <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">ssl</span><span class="p">:</span>
  658. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">shutdown</span><span class="p">()</span>
  659. <span class="k">else</span><span class="p">:</span>
  660. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">shutdown</span><span class="p">(</span><span class="n">socket</span><span class="o">.</span><span class="n">SHUT_RDWR</span><span class="p">)</span> <span class="c"># Make sure that it closes immediately.</span>
  661. <span class="n">Device</span><span class="o">.</span><span class="n">close</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
  662. <span class="k">except</span> <span class="ne">Exception</span><span class="p">,</span> <span class="n">ex</span><span class="p">:</span>
  663. <span class="k">pass</span>
  664. </div>
  665. <div class="viewcode-block" id="SocketDevice.write"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.write">[docs]</a> <span class="k">def</span> <span class="nf">write</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">data</span><span class="p">):</span>
  666. <span class="sd">&quot;&quot;&quot;</span>
  667. <span class="sd"> Writes data to the device.</span>
  668. <span class="sd"> :param data: The data to write.</span>
  669. <span class="sd"> :type data: str</span>
  670. <span class="sd"> :returns: The number of bytes sent.</span>
  671. <span class="sd"> :raises: CommError</span>
  672. <span class="sd"> &quot;&quot;&quot;</span>
  673. <span class="n">data_sent</span> <span class="o">=</span> <span class="bp">None</span>
  674. <span class="k">try</span><span class="p">:</span>
  675. <span class="n">data_sent</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">send</span><span class="p">(</span><span class="n">data</span><span class="p">)</span>
  676. <span class="k">if</span> <span class="n">data_sent</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
  677. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error writing to device.&#39;</span><span class="p">)</span>
  678. <span class="bp">self</span><span class="o">.</span><span class="n">on_write</span><span class="p">(</span><span class="n">data</span><span class="p">)</span>
  679. <span class="k">except</span> <span class="p">(</span><span class="n">SSL</span><span class="o">.</span><span class="n">Error</span><span class="p">,</span> <span class="n">socket</span><span class="o">.</span><span class="n">error</span><span class="p">),</span> <span class="n">err</span><span class="p">:</span>
  680. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error writing to device.&#39;</span><span class="p">,</span> <span class="n">err</span><span class="p">)</span>
  681. <span class="k">return</span> <span class="n">data_sent</span>
  682. </div>
  683. <div class="viewcode-block" id="SocketDevice.read"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.read">[docs]</a> <span class="k">def</span> <span class="nf">read</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  684. <span class="sd">&quot;&quot;&quot;</span>
  685. <span class="sd"> Reads a single character from the device.</span>
  686. <span class="sd"> :returns: The character read from the device.</span>
  687. <span class="sd"> :raises: CommError</span>
  688. <span class="sd"> &quot;&quot;&quot;</span>
  689. <span class="n">data</span> <span class="o">=</span> <span class="bp">None</span>
  690. <span class="k">try</span><span class="p">:</span>
  691. <span class="n">data</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">recv</span><span class="p">(</span><span class="mi">1</span><span class="p">)</span>
  692. <span class="k">except</span> <span class="n">socket</span><span class="o">.</span><span class="n">error</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  693. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error while reading from device: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  694. <span class="k">return</span> <span class="n">data</span>
  695. </div>
  696. <div class="viewcode-block" id="SocketDevice.read_line"><a class="viewcode-back" href="../../pyad2.html#pyad2.devices.SocketDevice.read_line">[docs]</a> <span class="k">def</span> <span class="nf">read_line</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">timeout</span><span class="o">=</span><span class="mf">0.0</span><span class="p">,</span> <span class="n">purge_buffer</span><span class="o">=</span><span class="bp">False</span><span class="p">):</span>
  697. <span class="sd">&quot;&quot;&quot;</span>
  698. <span class="sd"> Reads a line from the device.</span>
  699. <span class="sd"> :param timeout: The read timeout.</span>
  700. <span class="sd"> :type timeout: float</span>
  701. <span class="sd"> :param purge_buffer: Indicates whether to purge the buffer prior to reading.</span>
  702. <span class="sd"> :type purge_buffer: bool</span>
  703. <span class="sd"> :returns: The line read from the device.</span>
  704. <span class="sd"> :raises: CommError, TimeoutError</span>
  705. <span class="sd"> &quot;&quot;&quot;</span>
  706. <span class="k">if</span> <span class="n">purge_buffer</span><span class="p">:</span>
  707. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>
  708. <span class="k">def</span> <span class="nf">timeout_event</span><span class="p">():</span>
  709. <span class="n">timeout_event</span><span class="o">.</span><span class="n">reading</span> <span class="o">=</span> <span class="bp">False</span>
  710. <span class="n">timeout_event</span><span class="o">.</span><span class="n">reading</span> <span class="o">=</span> <span class="bp">True</span>
  711. <span class="n">got_line</span> <span class="o">=</span> <span class="bp">False</span>
  712. <span class="n">ret</span> <span class="o">=</span> <span class="bp">None</span>
  713. <span class="n">timer</span> <span class="o">=</span> <span class="bp">None</span>
  714. <span class="k">if</span> <span class="n">timeout</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">:</span>
  715. <span class="n">timer</span> <span class="o">=</span> <span class="n">threading</span><span class="o">.</span><span class="n">Timer</span><span class="p">(</span><span class="n">timeout</span><span class="p">,</span> <span class="n">timeout_event</span><span class="p">)</span>
  716. <span class="n">timer</span><span class="o">.</span><span class="n">start</span><span class="p">()</span>
  717. <span class="k">try</span><span class="p">:</span>
  718. <span class="k">while</span> <span class="n">timeout_event</span><span class="o">.</span><span class="n">reading</span><span class="p">:</span>
  719. <span class="n">buf</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="o">.</span><span class="n">recv</span><span class="p">(</span><span class="mi">1</span><span class="p">)</span>
  720. <span class="k">if</span> <span class="n">buf</span> <span class="o">!=</span> <span class="s">&#39;&#39;</span><span class="p">:</span>
  721. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">+=</span> <span class="n">buf</span>
  722. <span class="k">if</span> <span class="n">buf</span> <span class="o">==</span> <span class="s">&quot;</span><span class="se">\n</span><span class="s">&quot;</span><span class="p">:</span>
  723. <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">)</span> <span class="o">&gt;</span> <span class="mi">1</span><span class="p">:</span>
  724. <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">[</span><span class="o">-</span><span class="mi">2</span><span class="p">]</span> <span class="o">==</span> <span class="s">&quot;</span><span class="se">\r</span><span class="s">&quot;</span><span class="p">:</span>
  725. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">[:</span><span class="o">-</span><span class="mi">2</span><span class="p">]</span>
  726. <span class="c"># ignore if we just got \r\n with nothing else in the buffer.</span>
  727. <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">)</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">:</span>
  728. <span class="n">got_line</span> <span class="o">=</span> <span class="bp">True</span>
  729. <span class="k">break</span>
  730. <span class="k">else</span><span class="p">:</span>
  731. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span><span class="p">[:</span><span class="o">-</span><span class="mi">1</span><span class="p">]</span>
  732. <span class="k">except</span> <span class="n">socket</span><span class="o">.</span><span class="n">error</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  733. <span class="k">if</span> <span class="n">timer</span><span class="p">:</span>
  734. <span class="n">timer</span><span class="o">.</span><span class="n">cancel</span><span class="p">()</span>
  735. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error reading from device: {0}&#39;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">err</span><span class="p">)),</span> <span class="n">err</span><span class="p">)</span>
  736. <span class="k">else</span><span class="p">:</span>
  737. <span class="k">if</span> <span class="n">got_line</span><span class="p">:</span>
  738. <span class="n">ret</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span>
  739. <span class="bp">self</span><span class="o">.</span><span class="n">_buffer</span> <span class="o">=</span> <span class="s">&#39;&#39;</span>
  740. <span class="bp">self</span><span class="o">.</span><span class="n">on_read</span><span class="p">(</span><span class="n">ret</span><span class="p">)</span>
  741. <span class="k">if</span> <span class="n">timer</span><span class="p">:</span>
  742. <span class="k">if</span> <span class="n">timer</span><span class="o">.</span><span class="n">is_alive</span><span class="p">():</span>
  743. <span class="n">timer</span><span class="o">.</span><span class="n">cancel</span><span class="p">()</span>
  744. <span class="k">else</span><span class="p">:</span>
  745. <span class="k">raise</span> <span class="n">TimeoutError</span><span class="p">(</span><span class="s">&#39;Timeout while waiting for line terminator.&#39;</span><span class="p">)</span>
  746. <span class="k">return</span> <span class="n">ret</span>
  747. </div>
  748. <span class="k">def</span> <span class="nf">_init_ssl</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
  749. <span class="k">try</span><span class="p">:</span>
  750. <span class="n">ctx</span> <span class="o">=</span> <span class="n">SSL</span><span class="o">.</span><span class="n">Context</span><span class="p">(</span><span class="n">SSL</span><span class="o">.</span><span class="n">TLSv1_METHOD</span><span class="p">)</span>
  751. <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">ssl_key</span><span class="p">,</span> <span class="n">crypto</span><span class="o">.</span><span class="n">PKey</span><span class="p">):</span>
  752. <span class="n">ctx</span><span class="o">.</span><span class="n">use_privatekey</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">ssl_key</span><span class="p">)</span>
  753. <span class="k">else</span><span class="p">:</span>
  754. <span class="n">ctx</span><span class="o">.</span><span class="n">use_privatekey_file</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">ssl_key</span><span class="p">)</span>
  755. <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">ssl_certificate</span><span class="p">,</span> <span class="n">crypto</span><span class="o">.</span><span class="n">X509</span><span class="p">):</span>
  756. <span class="n">ctx</span><span class="o">.</span><span class="n">use_certificate</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">ssl_certificate</span><span class="p">)</span>
  757. <span class="k">else</span><span class="p">:</span>
  758. <span class="n">ctx</span><span class="o">.</span><span class="n">use_certificate_file</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">ssl_certificate</span><span class="p">)</span>
  759. <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">ssl_ca</span><span class="p">,</span> <span class="n">crypto</span><span class="o">.</span><span class="n">X509</span><span class="p">):</span>
  760. <span class="n">store</span> <span class="o">=</span> <span class="n">ctx</span><span class="o">.</span><span class="n">get_cert_store</span><span class="p">()</span>
  761. <span class="n">store</span><span class="o">.</span><span class="n">add_cert</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">ssl_ca</span><span class="p">)</span>
  762. <span class="k">else</span><span class="p">:</span>
  763. <span class="n">ctx</span><span class="o">.</span><span class="n">load_verify_locations</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">ssl_ca</span><span class="p">,</span> <span class="bp">None</span><span class="p">)</span>
  764. <span class="n">ctx</span><span class="o">.</span><span class="n">set_verify</span><span class="p">(</span><span class="n">SSL</span><span class="o">.</span><span class="n">VERIFY_PEER</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_verify_ssl_callback</span><span class="p">)</span>
  765. <span class="bp">self</span><span class="o">.</span><span class="n">_device</span> <span class="o">=</span> <span class="n">SSL</span><span class="o">.</span><span class="n">Connection</span><span class="p">(</span><span class="n">ctx</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">_device</span><span class="p">)</span>
  766. <span class="k">except</span> <span class="n">SSL</span><span class="o">.</span><span class="n">Error</span><span class="p">,</span> <span class="n">err</span><span class="p">:</span>
  767. <span class="k">raise</span> <span class="n">CommError</span><span class="p">(</span><span class="s">&#39;Error setting up SSL connection.&#39;</span><span class="p">,</span> <span class="n">err</span><span class="p">)</span>
  768. <span class="k">def</span> <span class="nf">_verify_ssl_callback</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">connection</span><span class="p">,</span> <span class="n">x509</span><span class="p">,</span> <span class="n">errnum</span><span class="p">,</span> <span class="n">errdepth</span><span class="p">,</span> <span class="n">ok</span><span class="p">):</span>
  769. <span class="k">return</span> <span class="n">ok</span></div>
  770. </pre></div>
  771. </div>
  772. </div>
  773. </div>
  774. <div class="sphinxsidebar">
  775. <div class="sphinxsidebarwrapper">
  776. <div id="searchbox" style="display: none">
  777. <h3>Quick search</h3>
  778. <form class="search" action="../../search.html" method="get">
  779. <input type="text" name="q" />
  780. <input type="submit" value="Go" />
  781. <input type="hidden" name="check_keywords" value="yes" />
  782. <input type="hidden" name="area" value="default" />
  783. </form>
  784. <p class="searchtip" style="font-size: 90%">
  785. Enter search terms or a module, class or function name.
  786. </p>
  787. </div>
  788. <script type="text/javascript">$('#searchbox').show(0);</script>
  789. </div>
  790. </div>
  791. <div class="clearer"></div>
  792. </div>
  793. <div class="related">
  794. <h3>Navigation</h3>
  795. <ul>
  796. <li class="right" style="margin-right: 10px">
  797. <a href="../../genindex.html" title="General Index"
  798. >index</a></li>
  799. <li class="right" >
  800. <a href="../../py-modindex.html" title="Python Module Index"
  801. >modules</a> |</li>
  802. <li><a href="../../index.html">pyad2 documentation</a> &raquo;</li>
  803. <li><a href="../index.html" >Module code</a> &raquo;</li>
  804. </ul>
  805. </div>
  806. <div class="footer">
  807. &copy; Copyright 2013, Author.
  808. Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.2b1.
  809. </div>
  810. </body>
  811. </html>