def get(self, request):
core = Core()
status, stats = core.db.getStats()
if status != DBStatus.s_ok:
logger.error("Could not get stats")
return Response({"error": "Could not get stats"}, status=500)
return Response(stats.__dict__)
def setUp(self):
self.core = Core()
self.core.db._populateIds()
user = DjangoUser.objects.create(username=self.username)
user.set_password(self.password)
user.save()
self.client = Client()
self.client.login(username=self.username, password=self.password)
def authenticate(self, login: str, password: str) -> bool:
from webforces.server.core import Core
core = Core()
status, user = core.db.getUserByLogin(login)
if status != DBStatus.s_ok:
user = User(0, login, "", "", "", [])
core.db.addUser(user)
logger.debug(f"Add new user to mongodb: {user}")
logger.debug(f"Logging in user: '******', password: {password}'")
def get(self, request):
core = Core()
status, users = core.db.getAllUsers()
if status != DBStatus.s_ok:
raise Exception("Could not get users")
data = {
"count": len(users),
}
return Response(data)
def setUp(self):
self.core = Core(validation=True)
self.core.db._populateIds()
user = DjangoUser.objects.create(username=self.username)
user.set_password(self.password)
user.is_staff = True
user.is_admin = True
user.is_superuser = True
user.save()
self.client = Client()
self.client.login(username=self.username, password=self.password)
def get_context_data(self, **kwargs):
if not self.request.user.is_superuser:
raise PermissionDenied
context = super().get_context_data(**kwargs)
core = Core()
status, stats = core.db.getStats()
logger.warning(stats)
if status != DBStatus.s_ok:
logger.error("Could not get stats")
raise Exception("Could not get stats")
context["stats"] = stats.__dict__
return context
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context["username"] = self.kwargs['user']
context["auth"] = self.request.user.is_authenticated
core = Core()
status, user = core.db.getUserByLogin(self.kwargs['user'])
if status != DBStatus.s_ok:
context["fullname"] = ""
else:
context[
"fullname"] = f"{user.first_name} {user.middle_name} {user.second_name}"
return context
def sign_up(request):
if request.method == 'POST':
form = UserCreationForm(request.POST)
if form.is_valid():
form.save()
username = form.cleaned_data.get('username')
raw_password = form.cleaned_data.get('password1')
user = authenticate(username=username, password=raw_password)
login(request, user)
Core().auth.register(username, raw_password)
return redirect('/')
else:
form = UserCreationForm()
return render(request, 'registration/signup.html', {'form': form})
def post(self, request, login):
core = Core()
status, user_db = core.db.getUserByLogin(login)
if status != DBStatus.s_ok:
return Response({"error": f"User does not exist: {status}"}, status=500)
user = request.data.dict()
user["user_id"] = user_db.user_id
user["login"] = login
user["algs_id"] = []
user = User.fromDict(user)
status = core.db.updUser(user)
if status != DBStatus.s_ok:
return Response({"error": f"Could not update user: {status}"}, status=500)
return Response({"success": f"User {login} was successfully updated"})
def get(self, request, login):
core = Core()
status, user = core.db.getUserByLogin(login)
if status != DBStatus.s_ok:
raise Http404("User does not exist")
data = {
"id": user.user_id,
"login": user.login,
"first_name": user.first_name,
"second_name": user.second_name,
"middle_name": user.middle_name,
}
return Response(data)
def log_in(request):
if request.method == 'POST':
form = AuthenticationForm(request.POST)
username = request.POST['username']
password = request.POST['password']
user = authenticate(username=username, password=password)
if user:
if user.is_active:
login(request, user)
Core().auth.authenticate(username, password)
return redirect(request.GET.get('next') or '/')
else:
messages.error(request, 'Incorrect username or password')
else:
form = AuthenticationForm()
return render(request, 'registration/login.html', {'form': form})
def get(self, request, title):
core = Core()
status, alg = core.db.getAlgByTitle(title)
if status != DBStatus.s_ok:
raise Exception("Could not get algorithm")
data = {
"alg_id": alg.alg_id,
"title": alg.title,
"author_id": alg.author_id,
"source": alg.source,
"tests_id": alg.tests_id,
"lang_id": alg.lang_id,
}
return Response(data)
import os
if os.environ.get("RUN_MAIN") == "true":
# Init webforces core
from webforces.server.core import Core
Core()
def _action(self, serializer):
logger.debug(self.request.data)
if 'username' in self.request.data.keys() and 'password' in self.request.data.keys():
Core().auth.authenticate(self.request.data['username'], self.request.data['password'])
return super()._action(serializer)
def setUp(self):
self.core = Core(validation=True)
self.core.db._populateIds()
def setUp(self):
self.core = Core()
self.core.db._populateIds()
self.client = Client()
def setUp(self):
self.core = Core(validation=True)
def test_core_is_proper_singletone(self):
core2 = Core(validation=True)
self.assertIs(core2, self.core)