update

1 files changed, 54 insertions, 5 deletions

diff --git a/seer.py b/seer.py
index 9226c39..9aa23fe 100755
--- a/seer.py
+++ b/seer.py
@@ -6,9 +6,16 @@ import code
 import readline
 import signal
 import sys
+import ccxt
+import pprint
 from lstm import lstm_type_1, lstm_type_2, lstm_type_3
 from marionette import marrionette_type_1
 from tfann import tfann_type_1
+from cnn import cnn_type_1
+import httplib2
+from googleapiclient.discovery import build
+import googleapiclient.http
+import oauth2client.client
 
 def SigHandler_SIGINT(signum, frame):
     print()
@@ -18,16 +25,55 @@ class Argparser(object):
     def __init__(self):
         parser = argparse.ArgumentParser()
         parser.add_argument("--which", type=str, help="which one to run")
-        parser.add_argument("--bool", action="store_true", help="bool", default=False)
+        parser.add_argument("--what", type=str, help="train or predict")
+        parser.add_argument("--pysrcupdate", type=str, nargs="+", help="name of source files to update on the drive")
+        parser.add_argument("--gpu", action="store_true", help="use gpu. if false will use cpu", default=False)
+        parser.add_argument("--test1", action="store_true", help="test switch 1", default=False)
+        parser.add_argument("--test2", action="store_true", help="test switch 2", default=False)
         parser.add_argument("--dbg", action="store_true", help="debug", default=False)
         self.args = parser.parse_args()
 
+def get_name_from_path(path):
+    path_pos = path.rfind("/")
+    if path_pos == -1:
+        return path
+    else:
+        return path[path_pos+1:]
+
+def g_drive_up(file_path, file_name, file_type, to_folder):
+    OAUTH2_SCOPE = "https://www.googleapis.com/auth/drive"
+    CLIENT_SECRETS = "./secret.json"
+    FILENAME = file_path
+    MIMETYPE = file_type
+    TITLE = file_name
+    DESCRIPTION = "a file"
+    flow = oauth2client.client.flow_from_clientsecrets(CLIENT_SECRETS, OAUTH2_SCOPE)
+    flow.redirect_uri = oauth2client.client.OOB_CALLBACK_URN
+    authorize_url = flow.step1_get_authorize_url()
+    print('Go to the following link in your browser: ' + authorize_url)
+    code = input('Enter verification code: ').strip()
+    credentials = flow.step2_exchange(code)
+    http = httplib2.Http()
+    credentials.authorize(http)
+    drive_service = build('drive', 'v3', http=http)
+    media_body = googleapiclient.http.MediaFileUpload(FILENAME, mimetype=MIMETYPE, resumable=True)
+    parent_dir = drive_service.files().list(q="mimeType='application/vnd.google-apps.folder' and name='"+to_folder+"'", fields="files(id, name)", spaces="drive").execute()
+    folder_id = str()
+    for file in parent_dir.get("files", []):
+        print(file.get("name") + "---" + file.get("id"))
+        folder_id = file.get("id")
+    #folder_id = "1FzAIXooboNvnAj4km5ujNporZp-ivro5"
+    body = {'name': TITLE, 'description': DESCRIPTION, 'parents': [folder_id]}
+    new_file = drive_service.files().create(body=body, media_body=media_body, fields="id").execute()
+    print(new_file.get("id"))
+    #pprint.pprint(new_file)
+
 def launch_ais(which):
     if which == "marionette": marrionette_type_1()
-    elif which == "lstm_type_1": lstm_type_1()
-    elif which == "lstm_type_2": lstm_type_2()
-    elif which == "lstm_type_3": lstm_type_3()
-    elif which == "cnn_type_1": pass
+    elif which == "lstm_type_1": lstm_type_1("ethereum", "ether")
+    elif which == "lstm_type_2": lstm_type_2("ethereum", "ether", 5, 20)
+    elif which == "lstm_type_3": lstm_type_3("ethereum", "ether", 5, 20)
+    elif which == "cnn_type_1": cnn_type_1()
     elif which == "tfann_type_1": tfann_type_1()
     else: pass
 
@@ -35,6 +81,9 @@ def launch_ais(which):
 def premain(argparser):
     signal.signal(signal.SIGINT, SigHandler_SIGINT)
     #here
+    if argparser.args.pysrcupdate:
+        for src in argparser.args.pysrcupdate:
+            g_drive_up(src, get_name_from_path(src), "text/python", "colab")
     launch_ais(argparser.args.which)
 
 def main():