def addPasswordInfo(user_name, password):
"""
Adds password for username/password combo in the URI.
a user with corresponding password must already exist
if not a 400 Bad Rquest is returned.
If the User is present and passwordInfo passes validation
then it is persisted and 200 OK is returned.
"""
try:
db = shelveWrapper(STORE_DB)
user = db.retriveUser(user_name, password)
if isinstance(user, User):
data = json.dumps(request.data)
data_str = json.loads(data)
UserDict = eval(data_str)
if len(UserDict) == 2 or len(UserDict) == 3:
db = shelveWrapper(STORE_DB)
reqKeys = UserDict.keys()
if len(reqKeys) == 2:
if 'name' in reqKeys and 'password' in reqKeys:
msg = db.writeEncrytptionData(user_name,
password,
UserDict['name'],
UserDict['password'],
'None')
else:
raise Exception
if len(reqKeys) == 3:
if (
'name' in reqKeys and
'password' in reqKeys and
'phrase' in reqKeys
):
msg = db.writeEncrytptionData(user_name,
password,
UserDict['name'],
UserDict['password'],
UserDict['phrase'])
else:
raise Exception
if msg == 'Password Info created successfully':
ret_data = {'msg': msg}
return jsonify(ret_data), 200
else:
ret_data = {'error': msg}
return jsonify(ret_data), 400
else:
raise Exception
else:
return_data = {'Error': {'msg': 'Invalid User info passed'}}
return jsonify(return_data), 404
except Exception, e:
print e
data_string = {'Error': 'Invalid data'}
return jsonify(data_string), 400
def addPasswordInfo(user_name, password):
"""
Adds password for username/password combo in the URI.
a user with corresponding password must already exist
if not a 400 Bad Rquest is returned.
If the User is present and passwordInfo passes validation
then it is persisted and 200 OK is returned.
"""
try:
db = shelveWrapper(STORE_DB)
user = db.retriveUser(user_name, password)
if isinstance(user, User):
data = json.dumps(request.data)
data_str = json.loads(data)
UserDict = eval(data_str)
if len(UserDict) == 2 or len(UserDict) == 3:
db = shelveWrapper(STORE_DB)
reqKeys = UserDict.keys()
if len(reqKeys) == 2:
if 'name' in reqKeys and 'password' in reqKeys:
msg = db.writeEncrytptionData(user_name, password,
UserDict['name'],
UserDict['password'],
'None')
else:
raise Exception
if len(reqKeys) == 3:
if ('name' in reqKeys and 'password' in reqKeys
and 'phrase' in reqKeys):
msg = db.writeEncrytptionData(user_name, password,
UserDict['name'],
UserDict['password'],
UserDict['phrase'])
else:
raise Exception
if msg == 'Password Info created successfully':
ret_data = {'msg': msg}
return jsonify(ret_data), 200
else:
ret_data = {'error': msg}
return jsonify(ret_data), 400
else:
raise Exception
else:
return_data = {'Error': {'msg': 'Invalid User info passed'}}
return jsonify(return_data), 404
except Exception, e:
print e
data_string = {'Error': 'Invalid data'}
return jsonify(data_string), 400
def getPasswordInfo(user_name, password, name):
"""
Gets passwordInfo with give name, the username and password in URI
must be present, if not 400 Bad Reqeust is returned.
If the give name is not persent 404 NOT found is returned
If the name exists then the 200 Ok is retruned with decryptd password.
"""
db = shelveWrapper(STORE_DB)
user = db.retriveUser(user_name, password)
if isinstance(user, User):
passwordInfo = db.retriePasswdInfo(user_name, password, name)
if isinstance(passwordInfo, AesEncryption):
passwd = AESDecryptionWrapper(user.key, passwordInfo.chiperText)
return_data = {
'result': 'found',
'phrase': passwordInfo.phrase,
'name': passwordInfo.name,
'password': passwd
}
return jsonify(return_data), 200
else:
return_data = {
'result': 'Not Found',
'msg': 'No password found with key ' + name
}
return jsonify(return_data), 404
else:
return_data = {'Error': {'msg': 'Invalid User info passed'}}
return jsonify(return_data), 404
def getPasswordInfo(user_name, password, name):
"""
Gets passwordInfo with give name, the username and password in URI
must be present, if not 400 Bad Reqeust is returned.
If the give name is not persent 404 NOT found is returned
If the name exists then the 200 Ok is retruned with decryptd password.
"""
db = shelveWrapper(STORE_DB)
user = db.retriveUser(user_name, password)
if isinstance(user, User):
passwordInfo = db.retriePasswdInfo(user_name, password, name)
if isinstance(passwordInfo, AesEncryption):
passwd = AESDecryptionWrapper(user.key,
passwordInfo.chiperText)
return_data = {'result': 'found', 'phrase': passwordInfo.phrase,
'name': passwordInfo.name,
'password': passwd}
return jsonify(return_data), 200
else:
return_data = {'result': 'Not Found',
'msg': 'No password found with key '+name}
return jsonify(return_data), 404
else:
return_data = {'Error': {'msg': 'Invalid User info passed'}}
return jsonify(return_data), 404
def getUser(user_name, password):
"""
Gets the user with user_name and password. If there is no User
with user_name password combo 404 Not Found is retruned.
If user exist, the the user is returned.
"""
db = shelveWrapper(STORE_DB)
user = db.retriveUser(user_name, password)
if isinstance(user, User):
return_data = {'result': 'found', 'phrase': user.phrase}
return jsonify(return_data), 200
else:
return_data = {'result': 'Not Found', 'msg': user}
return jsonify(return_data), 404
def getUser(user_name, password):
"""
Gets the user with user_name and password. If there is no User
with user_name password combo 404 Not Found is retruned.
If user exist, the the user is returned.
"""
db = shelveWrapper(STORE_DB)
user = db.retriveUser(user_name, password)
if isinstance(user, User):
return_data = {'result': 'found', 'phrase': user.phrase}
return jsonify(return_data), 200
else:
return_data = {'result': 'Not Found', 'msg': user}
return jsonify(return_data), 404
def addNewUser():
"""
This method is add/persist a new user to shelve db
If the user already exists Bad Request 400 is returned
If the user doesn't exist and passes user validation,
User is persisted and 200 OK is returned.
"""
try:
data = json.dumps(request.data)
data_str = json.loads(data)
UserDict = eval(data_str)
if len(UserDict) == 2 or len(UserDict) == 3:
db = shelveWrapper(STORE_DB)
reqKeys = UserDict.keys()
if len(reqKeys) == 2:
if 'password' in reqKeys and 'username' in reqKeys:
msg = db.writeUser(UserDict['username'],
UserDict['password'],
'None')
else:
raise Exception
if len(reqKeys) == 3:
if (
'password' in reqKeys and
'username' in reqKeys and
'phrase' in reqKeys
):
msg = db.writeUser(UserDict['username'],
UserDict['password'],
UserDict['phrase'])
else:
raise Exception
if msg == 'User Created Successfully':
ret_data = {'msg': msg}
return jsonify(ret_data), 200
else:
ret_data = {'error': msg}
return jsonify(ret_data), 400
else:
raise Exception
except Exception, e:
print e
data_string = {'Error': 'Invalid data'}
return jsonify(data_string), 400
def searchPasswordInfo(user_name, password, name, search):
"""
This method searches all passwords with give name and search string
There must be a existing username and password, if not 400 Bad request
is returned. The search string must be name or phrase, if not
400 Bad request is returned. If no password are found with search
criteria 404 not found is returned. Else a list of passwords qualifying
for search criteria is returned.
"""
db = shelveWrapper(STORE_DB)
if name == 'name' or name == 'phrase':
if len(search) <= 2:
return_data = {'Error': ' search string is too short'}
return jsonify(return_data), 400
search_result = db.retrivePasswordInfo_NameSearch(
user_name, password, name, search)
return jsonify(resut=search_result), 200
else:
return_data = {'Error': 'invalid search key'}
return jsonify(return_data), 400
def addNewUser():
"""
This method is add/persist a new user to shelve db
If the user already exists Bad Request 400 is returned
If the user doesn't exist and passes user validation,
User is persisted and 200 OK is returned.
"""
try:
data = json.dumps(request.data)
data_str = json.loads(data)
UserDict = eval(data_str)
if len(UserDict) == 2 or len(UserDict) == 3:
db = shelveWrapper(STORE_DB)
reqKeys = UserDict.keys()
if len(reqKeys) == 2:
if 'password' in reqKeys and 'username' in reqKeys:
msg = db.writeUser(UserDict['username'],
UserDict['password'], 'None')
else:
raise Exception
if len(reqKeys) == 3:
if ('password' in reqKeys and 'username' in reqKeys
and 'phrase' in reqKeys):
msg = db.writeUser(UserDict['username'],
UserDict['password'],
UserDict['phrase'])
else:
raise Exception
if msg == 'User Created Successfully':
ret_data = {'msg': msg}
return jsonify(ret_data), 200
else:
ret_data = {'error': msg}
return jsonify(ret_data), 400
else:
raise Exception
except Exception, e:
print e
data_string = {'Error': 'Invalid data'}
return jsonify(data_string), 400
def doSearch(store_db, userName, passwd, searchKey, searchString):
RESULT_LIST = []
funcs = [writer, reader]
nfuncs = range(len(funcs))
swrap = shelveWrapper(store_db)
user = swrap.retriveUser(userName, passwd)
if not isinstance(user, User):
if user == 'User Data not found':
return "{'error': ' invalid user info passed'}"
else:
dbHandle = shelve.open(swrap.shelveDB)
KeyList = dbHandle.keys()
nloops = len(KeyList)
q = Queue(QUEUE_SIZE)
threads = []
for i in nfuncs:
t = My_thread(funcs[i], (q, nloops, KeyList, dbHandle,
userName, searchKey, searchString,
RESULT_LIST), funcs[i].__name__)
threads.append(t)
for i in nfuncs:
threads[i].start()
for i in nfuncs:
threads[i].join()
if len(RESULT_LIST) == 0:
return"{'result: 'No passwords found after search'}"
else:
return_data = []
for data in RESULT_LIST:
passwd = AESDecryptionWrapper(user.key,
data.chiperText)
resDict = {'name': data.name, 'password': passwd,
'phrase': data.phrase}
return_data.append(resDict)
return return_data
def searchPasswordInfo(user_name, password, name, search):
"""
This method searches all passwords with give name and search string
There must be a existing username and password, if not 400 Bad request
is returned. The search string must be name or phrase, if not
400 Bad request is returned. If no password are found with search
criteria 404 not found is returned. Else a list of passwords qualifying
for search criteria is returned.
"""
db = shelveWrapper(STORE_DB)
if name == 'name' or name == 'phrase':
if len(search) <= 2:
return_data = {'Error': ' search string is too short'}
return jsonify(return_data), 400
search_result = db.retrivePasswordInfo_NameSearch(user_name,
password,
name,
search)
return jsonify(resut=search_result), 200
else:
return_data = {'Error': 'invalid search key'}
return jsonify(return_data), 400
def setUp(self):
self.db = shelveWrapper(TEST_DB)
def setUp(self):
self.db = shelveWrapper(TEST_DB)