Advertisement
Guest User

Untitled

a guest
Mar 25th, 2017
66
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.86 KB | None | 0 0
  1. class Tweet(models.Model):
  2. user = models.ForeignKey(User)
  3. encrypted_tweet = models.TextField()
  4. key = models.TextField()
  5. objects = TweetManager()
  6.  
  7. from cryptography.fernet import Fernet
  8.  
  9. def encrypt(request, id, postData):
  10. key = Fernet.generate_key()
  11. cipher_suite = Fernet(key)
  12. user = User.objects.get(id=id)
  13. binary_tweet = postData['tweet'].encode('UTF-8')
  14. encrypted_tweet = cipher_suite.encrypt(binary_tweet)
  15. tweet = Tweet.objects.create(user=user, encrypted_tweet=encrypted_tweet, key=key)
  16. return tweet.encrypted_tweet
  17.  
  18. def decrypt(request, search_string):
  19. binary_search_string = search_string.encode('UTF-8')
  20. tweet = Tweet.objects.get(encrypted_tweet=binary_search_string)
  21. key = tweet.key
  22. cipher_suite = Fernet(key)
  23. decrypted_tweet = cipher_suite.decrypt(binary_search_string)
  24. return decrypted_tweet
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement