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 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)
def test_contact_save(self):
contact = Contact(name="Irene",
email_address="*****@*****.**",
message="Hey.")
contact.save()
self.assertIsNotNone(contact.id)
