Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class Tweet(models.Model):
- user = models.ForeignKey(User)
- encrypted_tweet = models.TextField()
- key = models.TextField()
- objects = TweetManager()
- from cryptography.fernet import Fernet
- def encrypt(request, id, postData):
- key = Fernet.generate_key()
- cipher_suite = Fernet(key)
- user = User.objects.get(id=id)
- binary_tweet = postData['tweet'].encode('UTF-8')
- encrypted_tweet = cipher_suite.encrypt(binary_tweet)
- tweet = Tweet.objects.create(user=user, encrypted_tweet=encrypted_tweet, key=key)
- return tweet.encrypted_tweet
- def decrypt(request, search_string):
- binary_search_string = search_string.encode('UTF-8')
- tweet = Tweet.objects.get(encrypted_tweet=binary_search_string)
- key = tweet.key
- cipher_suite = Fernet(key)
- decrypted_tweet = cipher_suite.decrypt(binary_search_string)
- return decrypted_tweet
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement