def post_entry_add(request): engine = CryptoEngine(master_key=request.user.password) e = None attributes = set(['title', 'password', 'category']) if attributes <= set(request.POST.keys()): e = Entry(title=request.POST['title'], password=request.POST['password']) for k, v in request.POST.items(): if k == 'username': e.username = v elif k == 'url': e.url = v elif k == 'comment': e.comment = v elif k == 'expires': # date formated: m/d/Y e.expires = datetime.datetime.strptime(v, "%m/%d/%Y").date() elif k == 'category': c = Category.objects.filter(id=int(v)) if len(c) == 0: return HttpResponse(0) else: e.category_id = int(v) try: e.save() return HttpResponse(1) except: return HttpResponse(0) else: return HttpResponse(0)
def post(self, request, *args, **kwargs): if request.user.is_superuser: engine = CryptoEngine(master_key=request.user.password) form = EntryForm(request.POST) if form.is_valid(): entry = form.save(commit=False) entry.password = engine.encrypt(form.cleaned_data['password']) entry.save() messages.add_message( request, messages.INFO, u'New entry added: {}'.format(entry.title)) return redirect('home') else: form = EntryForm() return render(request, self.template_name, locals())
def get_search(request): engine = CryptoEngine(master_key=request.user.password) entries = None if 'title' in request.GET: search = str(request.GET['title']) entries = Entry.objects.filter(title__contains=search) elif 'category' in request.GET: search = str(request.GET['category']) categories = Category.objects.filter(title__contains=search) entries = [e for c in categories for e in c.entry_set.all()] else: entries = Entry.objects.all() # Decrypts the passwords for e in entries: e.password = engine.decrypt(e.password) response_data = [e.dict() for e in entries] return HttpResponse(json.dumps(response_data), content_type="application/json")
def decrypt(text, master_key): engine = CryptoEngine(master_key=master_key) return engine.decrypt(text)
def setUp(self): self.engine = CryptoEngine(master_key='mykeyisawesome!') self.block_length = 32 self.texts = CryptoEngineTest._generate_texts(100)