Advertisement
Guest User

Untitled

a guest
Mar 29th, 2020
107
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.92 KB | None | 0 0
  1. import os
  2. import tempfile
  3. import argparse
  4. import json
  5.  
  6. def put(key,value):
  7. data = get_data()
  8. data.setdefault(key,[]).append(value)
  9. with open(storage_path, 'w') as f:
  10. f.write(json.dumps(data))
  11. return
  12.  
  13. def get(key):
  14. data = get_data()
  15. return data.get(key)
  16.  
  17. def get_data():
  18. if not os.path.exists(storage_path):
  19. return {}
  20. with open(storage_path, 'r') as f:
  21. raw_data = f.read()
  22. if raw_data:
  23. return json.loads(raw_data)
  24. return {}
  25.  
  26.  
  27. parser = argparse.ArgumentParser()
  28. parser.add_argument('--key')
  29. parser.add_argument('--val')
  30. args = parser.parse_args()
  31.  
  32. storage_path = os.path.join(tempfile.gettempdir(), 'storage.data')
  33.  
  34.  
  35. if args.key is None:
  36. f = open(storage_path, 'w')
  37. f.truncate()
  38. elif args.val and args.key:
  39. put(args.key, args.val)
  40. elif args.key:
  41. print(get(args.key))
  42. else:
  43. print('Wrong command')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement