mirrors/dragonfly
1
0
Fork 0
mirror of https://github.com/dragonflydb/dragonfly.git synced 2025-05-11 10:25:47 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

feat: Add black formatter to the project (#1544)

Browse source 
Add black formatter and run it on pytests
This commit is contained in:
Kostas Kyrimis 2023-07-17 13:13:12 +03:00 • committed by GitHub
parent 9448220607
commit 7944af3c62
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
21 changed files with 793 additions and 566 deletions
Download patch file Download diff file Expand all files Collapse all files

40
tests/integration/generate_sets.py
View file

@ -12,40 +12,40 @@ def fill_set(args, redis: rclient.Redis):
for j in range(args.num):
token = uuid.uuid1().hex
# print(token)
key = f'USER_OTP:{token}'
key = f"USER_OTP:{token}"
arr = []
for i in range(30):
otp = ''.join(random.choices(
string.ascii_uppercase + string.digits, k=12))
otp = "".join(random.choices(string.ascii_uppercase + string.digits, k=12))
arr.append(otp)
redis.execute_command('sadd', key, *arr)
redis.execute_command("sadd", key, *arr)
def fill_hset(args, redis):
for j in range(args.num):
token = uuid.uuid1().hex
key = f'USER_INFO:{token}'
phone = f'555-999-{j}'
user_id = 'user' * 5 + f'-{j}'
redis.hset(key, 'phone', phone)
redis.hset(key, 'user_id', user_id)
redis.hset(key, 'login_time', time.time())
key = f"USER_INFO:{token}"
phone = f"555-999-{j}"
user_id = "user" * 5 + f"-{j}"
redis.hset(key, "phone", phone)
redis.hset(key, "user_id", user_id)
redis.hset(key, "login_time", time.time())
def main():
parser = argparse.ArgumentParser(description='fill hset entities')
parser = argparse.ArgumentParser(description="fill hset entities")
parser.add_argument("-p", type=int, help="redis port", dest="port", default=6380)
parser.add_argument("-n", type=int, help="number of keys", dest="num", default=10000)
parser.add_argument(
'-p', type=int, help='redis port', dest='port', default=6380)
parser.add_argument(
'-n', type=int, help='number of keys', dest='num', default=10000)
parser.add_argument(
'--type', type=str, choices=['hset', 'set'], help='set type', default='hset')
"--type", type=str, choices=["hset", "set"], help="set type", default="hset"
)
args = parser.parse_args()
redis = rclient.Redis(host='localhost', port=args.port, db=0)
if args.type == 'hset':
redis = rclient.Redis(host="localhost", port=args.port, db=0)
if args.type == "hset":
fill_hset(args, redis)
elif args.type == 'set':
elif args.type == "set":
fill_set(args, redis)
if __name__ == "__main__":
main()