From 6d317df46520bed7a7f21d326560452b8af6fb76 Mon Sep 17 00:00:00 2001
From: Roman Zeyde <roman.zeyde@gmail.com>
Date: Fri, 27 Mar 2015 09:48:51 +0300
Subject: [PATCH] scripts: fix lint warnings

---
 scripts/plot.py   | 13 +++++++++----
 scripts/record.py | 21 ++++++++++++---------
 2 files changed, 21 insertions(+), 13 deletions(-)

diff --git a/scripts/plot.py b/scripts/plot.py
index b2589c0..c4ec8f9 100755
--- a/scripts/plot.py
+++ b/scripts/plot.py
@@ -11,10 +11,12 @@ def spectrogram(t, x, Fs, NFFT=256):
     pylab.specgram(x, NFFT=NFFT, Fs=Fs, noverlap=NFFT/2,
                    cmap=pylab.cm.gist_heat)
 
-if __name__ == '__main__':
-    import sys
-    from amodem import common
-    from amodem.config import Configuration
+import sys
+from amodem import common
+from amodem.config import Configuration
+
+
+def main():
     config = Configuration()
 
     for fname in sys.argv[1:]:
@@ -25,3 +27,6 @@ if __name__ == '__main__':
         spectrogram(t, x, config.Fs)
 
     pylab.show()
+
+if __name__ == '__main__':
+    main()
diff --git a/scripts/record.py b/scripts/record.py
index dbf5600..51d38cb 100755
--- a/scripts/record.py
+++ b/scripts/record.py
@@ -4,12 +4,7 @@ from amodem import audio
 from amodem.config import Configuration
 
 
-def main():
-    p = argparse.ArgumentParser()
-    p.add_argument('-l', '--audio-library', default='libportaudio.so')
-    p.add_argument('filename')
-    args = p.parse_args()
-
+def run(args):
     config = Configuration()
     with open(args.filename, 'wb') as dst:
         print dst
@@ -21,8 +16,16 @@ def main():
                 dst.write(src.read(size))
 
 
-if __name__ == '__main__':
+def main():
+    p = argparse.ArgumentParser()
+    p.add_argument('-l', '--audio-library', default='libportaudio.so')
+    p.add_argument('filename')
+
     try:
-        main()
+        run(args=p.parse_args())
     except KeyboardInterrupt:
-        pass
+        return
+
+
+if __name__ == '__main__':
+    main()