App-Engine-Student-Connect/model.py at master · JoeyBoolean/App-Engine-Student-Connect · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
from google.appengine.ext import ndb

class Message(ndb.Model): #Guest
  first = ndb.StringProperty()
  last = ndb.StringProperty()
  userKey = ndb.KeyProperty()
  msg = ndb.StringProperty()
  time = ndb.DateTimeProperty(auto_now_add=True)

def AllMessages(): #AllGuests
  return Message.query()


def CourseMessages(course_key):
  msg = Message.query()
  print ('\n\n course_key\n')
  print(course_key)
  course = QuerySingleCourse(int(course_key))
  course = course.get()
  r = course.message_list
  print('\n Message List \n')
  print r
  return r

def UpdateMessage(id, first, last, msg): #UpdateGuest
  message = Message(id=id, first=first, last=last, msg=msg)
  message.put()
  return message


def InsertMessage(userKey, msg): #InsertGueset
  message = Message(userKey=userKey, msg=msg)
  r = message.put()
  return r


def DeleteMessage(id): #DeleteGuest
  key = ndb.Key(Message, id)
  key.delete()

class User(ndb.Model):
  username =ndb.StringProperty()
  name = ndb.StringProperty()
  password = ndb.StringProperty()
  logged_in = ndb.BooleanProperty()
  courses = ndb.KeyProperty(repeated = True)

def AllUsers():
  return User.query()

def QueryUser(id):
  user_key = ndb.Key(User, id)
  return user_key.get()

def CheckUser(username, password):
  print(username, password)
  users = User.query(User.username == username).fetch(1)
  # for user in users:
    # print(user.password)
  # user = users
  user = users.pop()
  print('Dis Shit')
  print(user)
  if user is None:
    return None
  # user = user_key.get()
  if(user.password == password):
    user.logged_in = True;
    user.put()
    return user
  return None

def UpdateUser(id, course_key):
  user_key = ndb.Key(User, id)
  user = user_key.get()
  course_list = user.courses
  course_list.append(course_key)
  user.courses = course_list
  user.put()

def InsertUser(username, name, password, course):
  course_list = []
  course_list.append(course)
  user = User(username=username, name=name, password=password, logged_in=True, courses=course_list)
  user.put()
  return user

class Course(ndb.Model):
  crn = ndb.IntegerProperty()
  name = ndb.StringProperty()
  message_list = ndb.KeyProperty(repeated = True)

def AddMessageToCourse(course_key, message_key):
  course = course_key.get()
  course_messages = course.message_list
  course_messages.append(message_key)
  course.message_list = course_messages
  course.put()

def InsertCourse(crn, name):
  course = Course(crn=int(crn), name=name, message_list=[])
  return course.put()

def QuerySingleCourse(course_key):
  course = ndb.Key(Course, course_key)
  return course

def QueryCourse(course_key):
  if(course_key is None):
    print('\nDoing this\n')
    course_key = AddCourseIfEmpty()
    # msgKey = InsertMessage()

  course_list = []
  for course in course_key:
    course_list.append(course.get())
  return course_list

def AddCourseIfEmpty():
  if( Course.query() ):
    print('\nhere\n')
    course = Course()
    course.crn = 00000
    course.name = 'Wichita State'
    course_key = course.put()
  else:
    course = Course.query()
    print('\nno here\n')
    print(course)
    print('\nno here\n')
    course_key = ndb.Key(Course, course)
    #print(course_key)

  return course_key