From c2397b9543d66d2ce2be680a25f9b006d23f528a Mon Sep 17 00:00:00 2001 From: Guillaume Simon Date: Fri, 8 Sep 2017 09:31:19 +0200 Subject: [PATCH 1/2] add example --- example.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100755 example.py diff --git a/example.py b/example.py new file mode 100755 index 0000000..bc2afa4 --- /dev/null +++ b/example.py @@ -0,0 +1,34 @@ +#!/usr/bin/env python3 +import wget +import zipfile +import pandas as pd +import os +from pdb import set_trace as bp +# +def dclean(dfiles): + for i, ival in enumerate(dfiles): + try: + os.remove(ival) + except OSError: + pass + return +# +zipfilename = 'occupancy_data.zip' +dfiles = ['datatest2.txt', 'datatest.txt', 'datatraining.txt'] +dfiles.append(zipfilename) +dclean(dfiles) +url = 'http://archive.ics.uci.edu/ml/machine-learning-databases/00357/'+zipfilename +filename = wget.download(url) +zip_ref = zipfile.ZipFile(filename, 'r') +zip_ref.extractall('./') +zip_ref.close() +df = pd.read_csv('datatraining.txt') +directory = 'asset/data' +if not os.path.exists(directory): + os.makedirs(directory) +df['time'] = df.index +df[['time','Temperature','Humidity']].to_csv(directory+'/sample.csv', index=False) +import train +if os.path.exists('asset'): + os.remove('asset') +dclean(dfiles) From fd20599769c42b9b091941eeabe7bd0662f5d499 Mon Sep 17 00:00:00 2001 From: Mark Date: Fri, 8 Sep 2017 09:49:58 +0200 Subject: [PATCH 2/2] add clean argument to example.py --- example.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/example.py b/example.py index bc2afa4..2781aad 100755 --- a/example.py +++ b/example.py @@ -3,9 +3,13 @@ import zipfile import pandas as pd import os +import shutil +import sys from pdb import set_trace as bp # def dclean(dfiles): + if os.path.exists('asset'): + shutil.rmtree('asset') for i, ival in enumerate(dfiles): try: os.remove(ival) @@ -17,6 +21,9 @@ def dclean(dfiles): dfiles = ['datatest2.txt', 'datatest.txt', 'datatraining.txt'] dfiles.append(zipfilename) dclean(dfiles) +if len(sys.argv) > 1: + if sys.argv[1] == 'clean': + sys.exit() url = 'http://archive.ics.uci.edu/ml/machine-learning-databases/00357/'+zipfilename filename = wget.download(url) zip_ref = zipfile.ZipFile(filename, 'r') @@ -29,6 +36,4 @@ def dclean(dfiles): df['time'] = df.index df[['time','Temperature','Humidity']].to_csv(directory+'/sample.csv', index=False) import train -if os.path.exists('asset'): - os.remove('asset') dclean(dfiles)