Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import os
- import tempfile
- import argparse
- import json
- def put(key,value):
- data = get_data()
- data.setdefault(key,[]).append(value)
- with open(storage_path, 'w') as f:
- f.write(json.dumps(data))
- return
- def get(key):
- data = get_data()
- return data.get(key)
- def get_data():
- if not os.path.exists(storage_path):
- return {}
- with open(storage_path, 'r') as f:
- raw_data = f.read()
- if raw_data:
- return json.loads(raw_data)
- return {}
- parser = argparse.ArgumentParser()
- parser.add_argument('--key')
- parser.add_argument('--val')
- args = parser.parse_args()
- storage_path = os.path.join(tempfile.gettempdir(), 'storage.data')
- if args.key is None:
- f = open(storage_path, 'w')
- f.truncate()
- elif args.val and args.key:
- put(args.key, args.val)
- elif args.key:
- print(get(args.key))
- else:
- print('Wrong command')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement