scripts: fix lint warnings

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()

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()