new function: write_schema

Author: Christopher Merrick

setup.cfg

@@ -0,0 +1,2 @@
+[metadata]
+description-file = README.md

stitchstream/__init__.py

@@ -1 +1 @@
-from .main import write_records, write_bookmark
+from .main import write_records, write_bookmark, write_schema

stitchstream/main.py

@@ -16,21 +16,27 @@ def _write_headers():
         _writeline('--')
         _headers_written = True
 
-def write_records(stream_name, schema, records):
+def write_records(stream_name, records):
     _write_headers()
     for record in records:
         _writeline(json.dumps({'type': 'RECORD',
                                'stream': stream_name,
-                               'schema': schema,
                                'record': record}))
 
+def write_schema(stream_name, schema):
+    _write_headers()
+    _writeline(json.dumps({'type': 'SCHEMA',
+                           'stream': stream_name,
+                           'schema': schema}))
+
 def write_bookmark(value):
     _write_headers()
     _writeline(json.dumps({'type': 'BOOKMARK',
                            'value': value}))
 
 
 if __name__ == "__main__":
+    write_schema('test',
+                 {'properties':{'id': {'type': 'string', 'key': True}}})
     write_records('test',
-                  {'properties':{'id': {'type': 'string', 'key': True}}},
                   [{'id': 'b'}, {'id':'d'}])