Mercurial > ~darius > hgwebdir.cgi > pyinst
annotate scpisock.py @ 73:ca5a822c550a
Fix for Python 3
Add getbin function to read large non-arrat data blocks (eg screen capture)
| author | Daniel O'Connor <doconnor@gsoft.com.au> |
|---|---|
| date | Wed, 25 Sep 2024 20:57:26 +0930 |
| parents | c6be52360c2f |
| children | b6ebe05f250f |
| rev | line source |
|---|---|
| 22 | 1 #!/usr/bin/env python |
| 2 | |
| 3 # Copyright (c) 2011 | |
| 4 # Daniel O'Connor <darius@dons.net.au>. All rights reserved. | |
| 5 # | |
| 6 # Redistribution and use in source and binary forms, with or without | |
| 7 # modification, are permitted provided that the following conditions | |
| 8 # are met: | |
| 9 # 1. Redistributions of source code must retain the above copyright | |
| 10 # notice, this list of conditions and the following disclaimer. | |
| 11 # 2. Redistributions in binary form must reproduce the above copyright | |
| 12 # notice, this list of conditions and the following disclaimer in the | |
| 13 # documentation and/or other materials provided with the distribution. | |
| 14 # | |
| 15 # THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND | |
| 16 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
| 17 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
| 18 # ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE | |
| 19 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
| 20 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | |
| 21 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |
| 22 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | |
| 23 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | |
| 24 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | |
| 25 # SUCH DAMAGE. | |
| 26 # | |
| 27 | |
| 28 import select | |
| 29 import socket | |
| 30 | |
| 31 # See this for more info | |
| 32 # http://www.staff.uni-bayreuth.de/~btp918/cmt2007/geraete/agilent_34410A_dmm/manual/SCPI_Sockets.pdf | |
| 33 # | |
| 34 # ToDo: Implement the SRQ & DCL channel (I don't have any supported hardware) | |
| 35 | |
| 36 SCPI_PORT = 5025 | |
| 37 | |
| 38 class SCPISockDevice(object): | |
|
28
c6be52360c2f
Handle default port better so passing None in picks the default.
Daniel O'Connor <darius@dons.net.au>
parents:
22
diff
changeset
|
39 def __init__(self, host, port = None): |
|
c6be52360c2f
Handle default port better so passing None in picks the default.
Daniel O'Connor <darius@dons.net.au>
parents:
22
diff
changeset
|
40 if port == None: |
|
c6be52360c2f
Handle default port better so passing None in picks the default.
Daniel O'Connor <darius@dons.net.au>
parents:
22
diff
changeset
|
41 port = SCPI_PORT |
| 22 | 42 self.sock = socket.create_connection((host, port)) |
| 43 | |
| 44 def flush(self): | |
| 45 while True: | |
| 46 r, w, x = select.select([self.sock], [], [], 0) | |
| 47 if len(r) == 0: | |
| 48 break | |
| 49 self.sock.recv(1024) | |
| 50 | |
| 51 def write(self, data): | |
| 52 trail = '' | |
| 53 if data[-1] != '\n': | |
| 54 trail = '\n' | |
| 55 | |
| 56 self.sock.send(data + trail) | |
| 57 | |
| 58 def read(self, timeout = None): | |
| 59 res = '' | |
| 60 if timeout == None: | |
| 61 timeout = 0.1 | |
| 62 | |
| 63 while True: | |
| 64 r, w, x = select.select([self.sock], [], [], timeout) | |
| 65 if len(r) == 0: | |
| 66 break | |
| 67 res = res + self.sock.recv(1024) | |
| 68 if res[-1] == '\n': | |
| 69 break | |
| 70 | |
| 71 return res.rstrip('\n') | |
| 72 |