def main(): generator=create_generator() i=0 for person in generator: i+=1 person.chanel_id=1 Session.add(person) if i==50: Session.commit() i=0 if i!=0: Session.commit()
def add_message(self): if request.method=='GET': return '' person_id=session.get('person_id') person_name=session.get('name') chanel_name=request.POST.get('chanel') chanel=self.chanel_q.filter_by(name=chanel_name).first() body=request.POST.get('body') to=request.POST.get('to') message=Message(body=body, date=get_date(), timestamp=int(time.time()), to=to, type='public', person_id=person_id, person_name=person_name,chanel_id=chanel.id) Session.add(message) Session.commit() c.body=body c.name=person_name c.date=get_date() # return render('ajax_add_message.mako') redirect('/chanel/'+chanel_name)
def signup(self): if request.method == "GET": redirect("/") email = self.form_result.get("email") name = self.form_result.get("name") password = self.form_result.get("password") result = self.person_q.filter_by(name=name).first() if result: return "This nickname is used by another one" result2 = self.person_q.filter_by(email=email).first() if result2: return "This email registered yet. Do u forget your password?" pers = Person(name=name, email=email, password=md5(password)) Session.add(pers) Session.commit() response.set_cookie("email", email) response.set_cookie("password", md5(password)) redirect("/chanel/main")
def create_chat(self): USER_CHATS_LIMIT=10 if request.method=='GET': return '' chat_name=request.POST.get('chat_name') result_query=self.chanel_q.filter_by(name=chat_name).first() if result_query: return 'This chanel name exists' owner_email=session.get('email') owner=self.person_q.filter_by(email=owner_email).first() if not owner: redirect('/auth/signin') chanels=self.chanel_q.filter_by(owner=owner.name).all() if len(chanels)>USER_CHATS_LIMIT: return 'You can`t create a chat' chanel=Chanel(name=chat_name, owner=owner.name, created='test') Session.add(chanel) Session.commit() redirect('/chanel/'+chat_name)