vpn integration
This commit is contained in:
@@ -1,9 +1,9 @@
|
||||
from django.core.management.base import BaseCommand
|
||||
|
||||
import json
|
||||
import base64
|
||||
import json
|
||||
import sys
|
||||
|
||||
from django.core.management.base import BaseCommand
|
||||
|
||||
from user.models import User
|
||||
|
||||
|
||||
@@ -26,7 +26,9 @@ class Command(BaseCommand):
|
||||
user_data = data["fields"]
|
||||
user_pk = data["pk"]
|
||||
if User.objects.filter(username__iexact=user_data["username"]).exists():
|
||||
old_new_users_maps[user_pk] = User.objects.filter(username__iexact=user_data["username"]).get()
|
||||
old_new_users_maps[user_pk] = User.objects.filter(
|
||||
username__iexact=user_data["username"]
|
||||
).get()
|
||||
else:
|
||||
old_new_users_maps[user_pk] = User.objects.create(
|
||||
email=user_data["email"],
|
||||
@@ -35,12 +37,14 @@ class Command(BaseCommand):
|
||||
is_superuser=user_data["is_superuser"],
|
||||
username=user_data["username"],
|
||||
password=user_data["password"],
|
||||
max_size=user_data["limit_size"]
|
||||
max_size=user_data["limit_size"],
|
||||
)
|
||||
old_friends[user_pk] = user_data["friends"]
|
||||
|
||||
for old_user, friends in old_friends.items():
|
||||
current_user = old_new_users_maps[old_user]
|
||||
for friend in friends:
|
||||
if not current_user.friends.filter(id=old_new_users_maps[friend].id).exists():
|
||||
if not current_user.friends.filter(
|
||||
id=old_new_users_maps[friend].id
|
||||
).exists():
|
||||
current_user.friends.add(old_new_users_maps[friend])
|
||||
|
||||
Reference in New Issue
Block a user