def prompt_add_new_contact(): """Add a new contact for user""" user_id = int(input("Please enter your user id: ")) contact_name = input("Please enter contact name: ") contact_phone_num = input("Please enter contact phone no: ") contact_email = input("Please enter contact email: ") new_contact = Contact(contact_name, contact_phone_num, contact_email, user_id) new_contact.save()
def contactData(req): back = {'status':'ok'} if req.META.has_key('HTTP_X_FORWARDED_FOR'): ip = request.META['HTTP_X_FORWARDED_FOR'] else: ip = req.META['REMOTE_ADDR'] q = req.GET or req.POST if Contact.objects.filter(ip=ip,check=0): back['status'] = 'error' back['msg'] = '您已提交信息,不能重复提交,我们会尽快跟您联系' else: c = Contact(name=q.get('name') or None,sex=q.get('sex') or None,email=q.get('email') or None,phone=q.get('phone') or None,advice=q.get('advice') or None,reply=q.get('reply') or None,ip=ip) c.save() back['msg'] = '您已提交信息,我们会尽快跟您联系' back['id'] = c.id return to_json(back)
def setUp(self): self.app = app self.app.register_blueprint(admin, url_prefix='/admin') self.test_client = self.app.test_client() self.db = db self.db.create_all() contact = Contact(name="Irene", email_address="*****@*****.**", message="Hi") contact.save() self.admin = User(email_address="*****@*****.**", password="******", admin=True) self.admin.save() self.not_admin = User(email_address="*****@*****.**", password="******", admin=False) self.not_admin.save()
def post_contact(): """ Adds contact. request body must specify as json: name - name of contact. email_address - email address of contact. message - a message. returns: json success or failure plus corresponding http status. """ contact_request = request.get_json() contact = Contact(**contact_request) success = contact.save() return jsonify({"contact": contact.id})
def test_contact_invalid_null_field(self): contact = Contact(name="Irene",message="Hey.") with self.assertRaises(IntegrityError): contact.save() self.assertIsNone(contact.id)
def test_contact_save(self): contact = Contact(name="Irene", email_address="*****@*****.**", message="Hey.") contact.save() self.assertIsNotNone(contact.id)
def test_contact_invalid_null_field(self): contact = Contact(name="Irene", message="Hey.") with self.assertRaises(IntegrityError): contact.save() self.assertIsNone(contact.id)