add helper function to automatically select native or RFC2217 implementation of Serial object

2 files changed, 16 insertions, 2 deletions

diff --git a/pyserial/serial/__init__.py b/pyserial/serial/__init__.py
index 6e160e0..acda005 100644
--- a/pyserial/serial/__init__.py
+++ b/pyserial/serial/__init__.py
@@ -24,3 +24,17 @@ else:
     else:
         raise Exception("Sorry: no implementation for your platform ('%s') available" % os.name)
 
+
+def serial_class_for_url(url, do_not_open=False, *args, **kwargs):
+    """Get a native or a RFC2217 implementation of the Serial class, depending
+    on port/url. The port is not opened when do_not_open is true, by default it is."""
+    if url.lower().startswith('rfc2217://'):
+        import rfc2217  # late import, so that users that don't use it don't have to load it
+        klass = rfc2217.Serial # RFC2217 implementation
+    else:
+        klass = Serial   # 'native' implementation
+    instance = klass(None, *args, **kwargs)
+    instance.port = url
+    if not do_not_open:
+        instance.open()
+    return instance
diff --git a/pyserial/serial/serialposix.py b/pyserial/serial/serialposix.py
index bbf7fb9..d9eec35 100644
--- a/pyserial/serial/serialposix.py
+++ b/pyserial/serial/serialposix.py
@@ -606,7 +606,7 @@ if __name__ == '__main__':
     s.flushInput()
     s.flushOutput()
     s.write('hello')
-    sys.stdio.write('%r\n' % s.read(5))
-    sys.stdio.write('%s\n' % s.inWaiting())
+    sys.stdout.write('%r\n' % s.read(5))
+    sys.stdout.write('%s\n' % s.inWaiting())
     del s