SkoolOS/CLI/skoolos.py

import sys
from urllib.parse import urlparse
import requests
from requests_oauthlib import OAuth2Session
from selenium import webdriver
import os.path
import time
import http.server
import socketserver
from threading import Thread
from werkzeug.urls import url_decode
import pprint
from PyInquirer import prompt, print_json
import json
import datetime
import os
import argparse
import webbrowser

client_id = r'QeZPBSKqdvWFfBv1VYTSv9iFGz5T9pVJtNUjbEr6'
client_secret = r'0Wl3hAIGY9SvYOqTOLUiLNYa4OlCgZYdno9ZbcgCT7RGQ8x2f1l2HzZHsQ7ijC74A0mrOhhCVeZugqAmOADHIv5fHxaa7GqFNtQr11HX9ySTw3DscKsphCVi5P71mlGY'
redirect_uri = 'http://localhost:8000/callback/'
token_url = 'https://ion.tjhsst.edu/oauth/token/'
scope = ["read"]
USER = ""
PWD = ""

def main():
    print("")
    print("░██████╗██╗░░██╗░█████╗░░█████╗░██╗░░░░░  ░█████╗░░██████╗")
    print("██╔════╝██║░██╔╝██╔══██╗██╔══██╗██║░░░░░  ██╔══██╗██╔════╝")
    print("╚█████╗░█████═╝░██║░░██║██║░░██║██║░░░░░  ██║░░██║╚█████╗░")
    print("░╚═══██╗██╔═██╗░██║░░██║██║░░██║██║░░░░░  ██║░░██║░╚═══██╗")
    print("██████╔╝██║░╚██╗╚█████╔╝╚█████╔╝███████╗  ╚█████╔╝██████╔╝")
    print("╚═════╝░╚═╝░░╚═╝░╚════╝░░╚════╝░╚══════╝  ░╚════╝░╚═════╝░")
    print("")

    if not (os.path.exists(".sprofile") or os.path.exists(".tprofile")):
        try:
            URL = "http://127.0.0.1:8000/api/"
            r = requests.get(url = URL)
        except:
            print("Run Django server on http://127.0.0.1:8000/ before continuing")
            sys.exit(0)

        input("Welcome to SkoolOS. Press any key to create an account")
        #webbrowser.open("http://127.0.0.1:8000/login", new=2)
        authenticate()
    else:
        try:
            f = open('.tprofile','r')
        except:
            f = open('.sprofile','r')
        data = json.loads(f.read())
        f.close()
        PWD = data['password']
        USER = data['username']
        print(data['username'])
        if(data['is_student']):
            studentCLI(USER, PWD)
        else:
            teacherCLI(USER, PWD)
        

    # while True:
    #     pass
def studentCLI(user, password):
    from CLI import student
    data = getUser(user, password, 'student')
    student = student.Student(data)
    carray = student.sclass.split(",")
    if(len(carray) == 1 and carray[0] == ""):
        print("No classes")
        return
        
    carray.append("Exit SkoolOS")
    courses = [
    {
        'type': 'list',
        'name': 'course',
        'choices':carray,
        'message': 'Select class: ',
    },
    ]
    course = prompt(courses)['course']
    print(course)
    if course == "Exit SkoolOS":
        student.exitCLI()
    else:
        student.viewClass(course)
        student.getAssignments(course,  100)

def teacherCLI(user, password):
    from CLI import teacher
    data = getUser(user, password, 'teacher')
    print(data)
    teacher = teacher.Teacher(data)
    # 1. make a class
    # 2. add studeents to an existing class
    # 3. Get progress logs on a student
    # 2. make an assignment for a class
    # 3. view student submissions for an assignment
    carray = teacher.classes
    carray.append("Exit SkoolOS")
    carray.append("Make New Class")
    courses = [
    {
        'type': 'list',
        'name': 'course',
        'choices':carray,
        'message': 'Select class: ',
    },
    ]
    course = prompt(courses)['course']
    if course == "Exit SkoolOS":
        teacher.exitCLI()
    if course == "Make New Class":
        questions = [
        {
            'type': 'input',
            'name': 'cname',
            'message': 'Class Name: ',
        },
        ]
        cname = prompt(questions)['cname']
        teacher.makeClass(cname)
        soption = ["1) Add individual student", "2) Add list of students through path", "3) Exit"]
        questions = [
        {
            'type': 'list',
            'choices':soption,
            'name': 'students',
            'message': 'Add list of students (input path): ',
        },        
        ]
        choice = prompt(questions)['students'].split(")")[0]
        if("1" == choice):
            s = input("Student name: ")
            teacher.addStudent(s, cname)
        if("2" == choice):
            p = input("Relativee Path: ")
            if(os.path.exists(p)):
                print(p + " does not exist.")

    else:
        print("Class: " + course)
        unconf = getDB("http://localhost:8000/api/classes/" + course)['unconfirmed']
        for s in unconf:
            teacher.addStudent()
        options = ['1) Request Student', "2) Add assignment", "3) View student information"]
        questions = [
        {
            'type': 'list',
            'name': 'course',
            'choices':options,
            'message': 'Select option: ',
        },
        ]
        option = prompt(questions)['course'].split(")")[0]
        if(option == '1'):
            soption = ["1) Add individual student", "2) Add list of students through path", "3) Exit"]
            questions = [
            {
                'type': 'list',
                'choices':soption,
                'name': 'students',
                'message': 'Add list of students (input path): ',
            },        
            ]
            schoice = prompt(questions)['students'].split(")")[0]
            if(schoice == '1'):
                questions = [
                {
                    'type': 'input',
                    'name': 'student',
                    'message': 'Student Name: ',
                },
                ]
                s = prompt(questions)['student']
                teacher.reqStudent(s, course)
            if(schoice == '2'):
                questions = [
                {
                    'type': 'input',
                    'name': 'path',
                    'message': 'Path: ',
                },
                ]
                path = prompt(questions)['path']
                while(not os.path.exists(path)):
                    if(path == 'N'):
                        sys.exit(0)
                    print(path + " is not a valid path")
                    path = input("Enter file path ('N' to exit): ")
                f = open(path, 'r')
                students = f.read().splitlines()
                teacher.reqAddStudentList(students, course)
            else:
                sys.exit()
        if(option == '2'):
            nlist = os.listdir(teacher.username + "/" + course)
            alist = getDB("http://localhost:8000/api/classes/" + course)['assignments']
            print(nlist)
            tlist = []
            b = True
            for n in nlist:
                b = True
                print(teacher.username + "/" + course + "/" + n)
                for a  in alist:
                    if(n in a or n == a):
                        #print("Assignments: " + n)
                        b = False
                if(not os.path.isdir(teacher.username + "/" + course + "/" + n)):
                    b = False
                if(b):
                    tlist.append(n)


            nlist = tlist
            if(len(nlist) == 0):
                print("No new assignments found")
                sys.exit(0)
            questions = [
            {
                'type': 'list',
                'choices':nlist,
                'name': 'assignment',
                'message': 'Select new assignment: ',
            },        
            ]
            ass = prompt(questions)['assignment']
            apath = teacher.username + "/" + course + "/" + ass
            
            teacher.addAssignment(apath, course, due)


def getUser(ion_user, password, utype):
        if('student' in utype):
            URL = "http://127.0.0.1:8000/api/students/" + ion_user + "/"
        else:
            URL = "http://127.0.0.1:8000/api/teachers/" + ion_user + "/"
        r = requests.get(url = URL, auth=(ion_user,password)) 
        print(r.json())
        if(r.status_code == 200):
            data = r.json() 
            print(200)
            return data
        elif(r.status_code == 404):
            print("Make new account!")
            return None
        elif(r.status_code == 403):
            print("Invalid username/password")
            return None
        else:
            print(r.status_code) 
            return None
def patchDB(data, url):
    r = requests.patch(url = url, data=data, auth=('raffukhondaker','hackgroup1'))
    print("PATH:" + str(r.status_code))
    return(r.json())

def getDB(url):
    r = requests.get(url = url, auth=('raffukhondaker','hackgroup1')) 
    print("GET:" + str(r.status_code))
    return(r.json())

def postDB(data, url):
    r = requests.post(url = url, data=data, auth=('raffukhondaker','hackgroup1')) 
    print("POST:" + str(r.status_code))
    return(r.json())

def putDB(data, url):
    r = requests.put(url = url, data=data, auth=('raffukhondaker','hackgroup1'))
    print("PUT:" + str(r.status_code))
    return(r.json())

def delDB(url):
    r = requests.delete(url = url, auth=('raffukhondaker','hackgroup1'))
    print("DELETE:" + str(r.status_code))
    return None

def makePass():
    questions = [
    {
        'type': 'password',
        'name': 'pwd',
        'message': 'Enter SkoolOS Password (NOT ION PASSWORD): ',
    },
    ]
    pwd = prompt(questions)['pwd']
    while(len(pwd) < 7):
        print("Password too short (Must be over 6 characters)")
        pwd = prompt(questions)['pwd']
    conf = [
    {
        'type': 'password',
        'name': 'pwd',
        'message': 'Re-enter password: ',
    },
    ]
    pwd2 = prompt(conf)['pwd']
    while(not pwd == pwd2):
        print("Passwords do not match.")
        pwd2 = prompt(conf)['pwd']
    else:
        print("PASSWORD SAVED")
        return pwd

def authenticate():
    oauth = OAuth2Session(client_id=client_id, redirect_uri=redirect_uri, scope=scope)
    authorization_url, state = oauth.authorization_url("https://ion.tjhsst.edu/oauth/authorize/")

    cdir = os.getcwd()
    #Linux: chromdriver-linux
    #Macos: chromdriver-mac
    #Windows: chromdriver.exe
    if('CLI' in os.getcwd()):
        path = os.path.join(os.getcwd(), '../','chromedriver/chromedriver-mac')
    else:
        path = os.path.join(os.getcwd(), 'chromedriver/chromedriver-mac')

    browser = webdriver.Chrome(path)

    browser.get("localhost:8000/login")

    # while "http://localhost:8000/callback/?code" not in browser.current_url:
    #     time.sleep(0.25)

    url = browser.current_url
    gets = url_decode(url.replace("http://localhost:8000/login/?", ""))
    while "http://localhost:8000/login/?username=" not in browser.current_url and (not browser.current_url == "http://localhost:8000/"): #http://localhost:8000/
        time.sleep(0.25)

    url = browser.current_url
    gets = url_decode(url.replace("http://localhost:8000/login/?username=", ""))
    # code = gets.get("code")
    # if state == gets.get("state"):
    #     state = gets.get("state")
    #     print("states good")
    browser.quit()
    questions = [
    {
        'type': 'input',
        'name': 'username',
        'message': 'Enter SkoolOS Username (Same as ION Username): ',
    },
    {
        'type': 'password',
        'name': 'pwd',
        'message': 'Enter SkoolOS Password (NOT ION PASSWORD): ',
    },
    ]
    data =prompt(questions) 
    pwd = data['pwd']
    user = data['username']
    r = requests.get(url = "http://localhost:8000/api/", auth=(user,pwd)) 
    while(r.status_code != 200):
        print("INCORRECT LOGIN CREDENTIALS")
        r = requests.get(url = "http://localhost:8000/api/", auth=(user,pwd)) 
        data =prompt(questions) 
        pwd = data['pwd']
        user = data['username']
        print(r.status_code)
    r = requests.get(url = "http://localhost:8000/api/students/" + user + "/", auth=(user,pwd)) 
    is_student = False
    if(r.status_code == 200):
        is_student = True
        print("Welcome, student " + user)
        r = requests.get(url = "http://localhost:8000/api/students/" + user + "/", auth=(user,pwd))
        profile = r.json()
        username = profile['ion_user']
        grade = profile['grade']
        profile = {
            'username':username,
            'grade':grade,
            'is_student':is_student,
            'password':pwd,
        }
        profileFile = open(".sprofile", "w")
        profileFile.write(json.dumps(profile))
        profileFile.close()

    else:
        print("Welcome, teacher " + user)
        r = requests.get(url = "http://localhost:8000/api/teachers/" + user + "/", auth=(user,pwd))
        profile = r.json()
        username = profile['ion_user']
        profile = {
            'username':username,
            'is_student':is_student,
            'password':pwd,
        }
        profileFile = open(".tprofile", "w")
        profileFile.write(json.dumps(profile))
        profileFile.close()

    sys.exit

def create_server():
    port = 8000
    handler = http.server.SimpleHTTPRequestHandler
    httpd = socketserver.TCPServer(("", port), handler)
    print("serving at port:" + str(port))
    httpd.serve_forever()

if __name__ == "__main__":
    main()