mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2026-02-04 14:16:21 +01:00
Compare commits
154 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
da93940e13 | ||
|
|
f53b43d9dc | ||
|
|
497a167ca7 | ||
|
|
852c5a3037 | ||
|
|
90f6777a7a | ||
|
|
a8d890abaf | ||
|
|
79379b444c | ||
|
|
ace1de5c68 | ||
|
|
5d6128e9ea | ||
|
|
131cc5ed0a | ||
|
|
9297bed43f | ||
|
|
b2fddc683a | ||
|
|
9af634d16a | ||
|
|
a68163caa3 | ||
|
|
00770fd034 | ||
|
|
01164777ed | ||
|
|
00f5e3e1e0 | ||
|
|
8b2527f408 | ||
|
|
b7500e4e4e | ||
|
|
4f4bd0c419 | ||
|
|
8ae4e02012 | ||
|
|
cc9a07197d | ||
|
|
f18dd1029b | ||
|
|
fd8d43571a | ||
|
|
13e88492f1 | ||
|
|
38df580a56 | ||
|
|
ba39318313 | ||
|
|
d8c6035405 | ||
|
|
2ef3da916b | ||
|
|
d32d8b26ce | ||
|
|
f348b1a34c | ||
|
|
86aaa3edda | ||
|
|
26017056c7 | ||
|
|
e39a3c072b | ||
|
|
827291dda4 | ||
|
|
ea8958ccc3 | ||
|
|
20554df857 | ||
|
|
750f43eaf0 | ||
|
|
09cf28ec9f | ||
|
|
b61746b3cb | ||
|
|
22c22fafeb | ||
|
|
577c4395c4 | ||
|
|
d241f476f7 | ||
|
|
5832ed0c30 | ||
|
|
bd9ea225be | ||
|
|
4a575dd70c | ||
|
|
b80ee16a7c | ||
|
|
c888371e6c | ||
|
|
8de2c3bfcb | ||
|
|
6688f73565 | ||
|
|
7d929cb6e2 | ||
|
|
72740b9e4d | ||
|
|
f7d279fa16 | ||
|
|
ff7c9c48f3 | ||
|
|
d11832913d | ||
|
|
724e0e83f2 | ||
|
|
333f091f1a | ||
|
|
cfbb0b993a | ||
|
|
582b6754a4 | ||
|
|
7767c46bf4 | ||
|
|
bf34cef896 | ||
|
|
c085ec6860 | ||
|
|
5f5d0316b2 | ||
|
|
f9ec64c3ad | ||
|
|
0dfd0ad4b0 | ||
|
|
e88e11b9ba | ||
|
|
7a2a79ca7b | ||
|
|
4c0683c484 | ||
|
|
dfe62db8ee | ||
|
|
025c824fbb | ||
|
|
930c5d7c7a | ||
|
|
8b8dcc0127 | ||
|
|
4ad8e88bd8 | ||
|
|
89d4640e92 | ||
|
|
60b12bad61 | ||
|
|
2595fa5c51 | ||
|
|
3e487e5f13 | ||
|
|
b6d6c68e54 | ||
|
|
49548d6f9f | ||
|
|
8faadc23b0 | ||
|
|
20da1ebfab | ||
|
|
03305c72c7 | ||
|
|
a636fd1cf0 | ||
|
|
d8797a8dc6 | ||
|
|
363e18e15d | ||
|
|
982cac8c43 | ||
|
|
fabf64b838 | ||
|
|
e3e6ebe953 | ||
|
|
7ad9b52546 | ||
|
|
abb5090d63 | ||
|
|
52ae05d057 | ||
|
|
3cd216d119 | ||
|
|
581edc0a38 | ||
|
|
f17ebbede6 | ||
|
|
2bd2c09c23 | ||
|
|
c1cf859ec9 | ||
|
|
1ebf864998 | ||
|
|
0948e34e48 | ||
|
|
f9a1ea9c83 | ||
|
|
d02a8ebc1b | ||
|
|
5c2625b648 | ||
|
|
13174d006e | ||
|
|
2c59cc4cc3 | ||
|
|
41c81d3226 | ||
|
|
2ec7d3b4d9 | ||
|
|
b607b73598 | ||
|
|
f52791cd1f | ||
|
|
d829facbd4 | ||
|
|
44ac3a9ff2 | ||
|
|
2054a76353 | ||
|
|
4b073c3161 | ||
|
|
527c016c72 | ||
|
|
23849b9f42 | ||
|
|
5d4277adf0 | ||
|
|
0f479d525e | ||
|
|
4aae85a70f | ||
|
|
f0ec16be4e | ||
|
|
92f6d8b98d | ||
|
|
0a48436937 | ||
|
|
f4c5dc77d8 | ||
|
|
8656d1c972 | ||
|
|
1b81f5cfe8 | ||
|
|
a4723c5942 | ||
|
|
b31521c758 | ||
|
|
6dc96bb348 | ||
|
|
2da78f7793 | ||
|
|
e8d54a1d11 | ||
|
|
f79e764439 | ||
|
|
803b659fce | ||
|
|
537987398d | ||
|
|
cdcf65cb52 | ||
|
|
5fcb092151 | ||
|
|
1fd708e765 | ||
|
|
a19302babc | ||
|
|
18a627b01e | ||
|
|
a6b340c179 | ||
|
|
2fe1de1c97 | ||
|
|
0072983b4f | ||
|
|
e1e87edc74 | ||
|
|
eddb5480e9 | ||
|
|
f3c28ea933 | ||
|
|
829f8ba0a4 | ||
|
|
f49ca2bccb | ||
|
|
9c71a8d9a3 | ||
|
|
0032f91525 | ||
|
|
7f3492f978 | ||
|
|
2b9110e417 | ||
|
|
98619a0eb8 | ||
|
|
7a4aa05c39 | ||
|
|
5b26757662 | ||
|
|
ba597cd2ad | ||
|
|
5992ddd48e | ||
|
|
8486b95917 | ||
|
|
bb15de6d1a |
@@ -22,3 +22,7 @@ indent_style = tab
|
|||||||
|
|
||||||
[*.bat]
|
[*.bat]
|
||||||
indent_style = tab
|
indent_style = tab
|
||||||
|
|
||||||
|
[{Dockerfile,*.dockerfile}]
|
||||||
|
indent_style = space
|
||||||
|
indent_size = 4
|
||||||
|
|||||||
2
.gitignore
vendored
2
.gitignore
vendored
@@ -38,7 +38,6 @@ htmlcov/
|
|||||||
.tox/
|
.tox/
|
||||||
.coverage
|
.coverage
|
||||||
.cache
|
.cache
|
||||||
nosetests.xml
|
|
||||||
coverage.xml
|
coverage.xml
|
||||||
|
|
||||||
# Translations
|
# Translations
|
||||||
@@ -77,3 +76,4 @@ celerybeat-schedule
|
|||||||
.flake8
|
.flake8
|
||||||
.pylintrc
|
.pylintrc
|
||||||
Makefile
|
Makefile
|
||||||
|
.isort.cfg
|
||||||
|
|||||||
120
.gitlab-ci.yml
120
.gitlab-ci.yml
@@ -1,8 +1,15 @@
|
|||||||
|
.only-default: &only-default
|
||||||
|
only:
|
||||||
|
- master
|
||||||
|
- branches
|
||||||
|
- merge_requests
|
||||||
|
|
||||||
stages:
|
stages:
|
||||||
- pre-commit
|
- pre-commit
|
||||||
- gitlab
|
- gitlab
|
||||||
- test
|
- test
|
||||||
- deploy
|
- deploy
|
||||||
|
- docker
|
||||||
|
|
||||||
include:
|
include:
|
||||||
- template: Dependency-Scanning.gitlab-ci.yml
|
- template: Dependency-Scanning.gitlab-ci.yml
|
||||||
@@ -15,6 +22,7 @@ before_script:
|
|||||||
- pip install wheel tox
|
- pip install wheel tox
|
||||||
|
|
||||||
pre-commit-check:
|
pre-commit-check:
|
||||||
|
<<: *only-default
|
||||||
stage: pre-commit
|
stage: pre-commit
|
||||||
image: python:3.6-buster
|
image: python:3.6-buster
|
||||||
variables:
|
variables:
|
||||||
@@ -39,6 +47,7 @@ dependency_scanning:
|
|||||||
- pip install wheel tox
|
- pip install wheel tox
|
||||||
|
|
||||||
test-3.7-core:
|
test-3.7-core:
|
||||||
|
<<: *only-default
|
||||||
image: python:3.7-bullseye
|
image: python:3.7-bullseye
|
||||||
script:
|
script:
|
||||||
- tox -e py37-core
|
- tox -e py37-core
|
||||||
@@ -48,6 +57,7 @@ test-3.7-core:
|
|||||||
cobertura: coverage.xml
|
cobertura: coverage.xml
|
||||||
|
|
||||||
test-3.8-core:
|
test-3.8-core:
|
||||||
|
<<: *only-default
|
||||||
image: python:3.8-bullseye
|
image: python:3.8-bullseye
|
||||||
script:
|
script:
|
||||||
- tox -e py38-core
|
- tox -e py38-core
|
||||||
@@ -57,6 +67,7 @@ test-3.8-core:
|
|||||||
cobertura: coverage.xml
|
cobertura: coverage.xml
|
||||||
|
|
||||||
test-3.9-core:
|
test-3.9-core:
|
||||||
|
<<: *only-default
|
||||||
image: python:3.9-bullseye
|
image: python:3.9-bullseye
|
||||||
script:
|
script:
|
||||||
- tox -e py39-core
|
- tox -e py39-core
|
||||||
@@ -65,7 +76,29 @@ test-3.9-core:
|
|||||||
reports:
|
reports:
|
||||||
cobertura: coverage.xml
|
cobertura: coverage.xml
|
||||||
|
|
||||||
|
test-3.10-core:
|
||||||
|
<<: *only-default
|
||||||
|
image: python:3.10-bullseye
|
||||||
|
script:
|
||||||
|
- tox -e py310-core
|
||||||
|
artifacts:
|
||||||
|
when: always
|
||||||
|
reports:
|
||||||
|
cobertura: coverage.xml
|
||||||
|
|
||||||
|
test-3.11-core:
|
||||||
|
<<: *only-default
|
||||||
|
image: python:3.11-rc-bullseye
|
||||||
|
script:
|
||||||
|
- tox -e py311-core
|
||||||
|
artifacts:
|
||||||
|
when: always
|
||||||
|
reports:
|
||||||
|
cobertura: coverage.xml
|
||||||
|
allow_failure: true
|
||||||
|
|
||||||
test-3.7-all:
|
test-3.7-all:
|
||||||
|
<<: *only-default
|
||||||
image: python:3.7-bullseye
|
image: python:3.7-bullseye
|
||||||
script:
|
script:
|
||||||
- tox -e py37-all
|
- tox -e py37-all
|
||||||
@@ -75,6 +108,7 @@ test-3.7-all:
|
|||||||
cobertura: coverage.xml
|
cobertura: coverage.xml
|
||||||
|
|
||||||
test-3.8-all:
|
test-3.8-all:
|
||||||
|
<<: *only-default
|
||||||
image: python:3.8-bullseye
|
image: python:3.8-bullseye
|
||||||
script:
|
script:
|
||||||
- tox -e py38-all
|
- tox -e py38-all
|
||||||
@@ -84,6 +118,7 @@ test-3.8-all:
|
|||||||
cobertura: coverage.xml
|
cobertura: coverage.xml
|
||||||
|
|
||||||
test-3.9-all:
|
test-3.9-all:
|
||||||
|
<<: *only-default
|
||||||
image: python:3.9-bullseye
|
image: python:3.9-bullseye
|
||||||
script:
|
script:
|
||||||
- tox -e py39-all
|
- tox -e py39-all
|
||||||
@@ -92,9 +127,30 @@ test-3.9-all:
|
|||||||
reports:
|
reports:
|
||||||
cobertura: coverage.xml
|
cobertura: coverage.xml
|
||||||
|
|
||||||
|
test-3.10-all:
|
||||||
|
<<: *only-default
|
||||||
|
image: python:3.10-bullseye
|
||||||
|
script:
|
||||||
|
- tox -e py310-all
|
||||||
|
artifacts:
|
||||||
|
when: always
|
||||||
|
reports:
|
||||||
|
cobertura: coverage.xml
|
||||||
|
|
||||||
|
test-3.11-all:
|
||||||
|
<<: *only-default
|
||||||
|
image: python:3.11-rc-bullseye
|
||||||
|
script:
|
||||||
|
- tox -e py311-all
|
||||||
|
artifacts:
|
||||||
|
when: always
|
||||||
|
reports:
|
||||||
|
cobertura: coverage.xml
|
||||||
|
allow_failure: true
|
||||||
|
|
||||||
deploy_production:
|
deploy_production:
|
||||||
stage: deploy
|
stage: deploy
|
||||||
image: python:3.9-bullseye
|
image: python:3.10-bullseye
|
||||||
|
|
||||||
before_script:
|
before_script:
|
||||||
- pip install twine wheel
|
- pip install twine wheel
|
||||||
@@ -105,3 +161,65 @@ deploy_production:
|
|||||||
|
|
||||||
rules:
|
rules:
|
||||||
- if: $CI_COMMIT_TAG
|
- if: $CI_COMMIT_TAG
|
||||||
|
|
||||||
|
build-image:
|
||||||
|
before_script: []
|
||||||
|
image: docker:20.10.10
|
||||||
|
stage: docker
|
||||||
|
services:
|
||||||
|
- docker:20.10.10-dind
|
||||||
|
script: |
|
||||||
|
CURRENT_DATE=$(echo $CI_COMMIT_TIMESTAMP | head -c 10 | tr -d -)
|
||||||
|
IMAGE_TAG=$CI_REGISTRY_IMAGE/auth:$CURRENT_DATE-$CI_COMMIT_SHORT_SHA
|
||||||
|
CURRENT_TAG=$CI_REGISTRY_IMAGE/auth:$CI_COMMIT_TAG
|
||||||
|
MINOR_TAG=$CI_REGISTRY_IMAGE/auth:$(echo $CI_COMMIT_TAG | cut -d '.' -f 1-2)
|
||||||
|
MAJOR_TAG=$CI_REGISTRY_IMAGE/auth:$(echo $CI_COMMIT_TAG | cut -d '.' -f 1)
|
||||||
|
LATEST_TAG=$CI_REGISTRY_IMAGE/auth:latest
|
||||||
|
|
||||||
|
docker login -u $CI_REGISTRY_USER -p $CI_REGISTRY_PASSWORD $CI_REGISTRY
|
||||||
|
docker build . -t $IMAGE_TAG -f docker/Dockerfile --build-arg AUTH_VERSION=$(echo $CI_COMMIT_TAG | cut -c 2-)
|
||||||
|
docker tag $IMAGE_TAG $CURRENT_TAG
|
||||||
|
docker tag $IMAGE_TAG $MINOR_TAG
|
||||||
|
docker tag $IMAGE_TAG $MAJOR_TAG
|
||||||
|
docker tag $IMAGE_TAG $LATEST_TAG
|
||||||
|
docker image push --all-tags $CI_REGISTRY_IMAGE/auth
|
||||||
|
rules:
|
||||||
|
- if: $CI_COMMIT_TAG
|
||||||
|
|
||||||
|
build-image-dev:
|
||||||
|
before_script: []
|
||||||
|
image: docker:20.10.10
|
||||||
|
stage: docker
|
||||||
|
services:
|
||||||
|
- docker:20.10.10-dind
|
||||||
|
script: |
|
||||||
|
CURRENT_DATE=$(echo $CI_COMMIT_TIMESTAMP | head -c 10 | tr -d -)
|
||||||
|
IMAGE_TAG=$CI_REGISTRY_IMAGE/auth:$CURRENT_DATE-$CI_COMMIT_BRANCH-$CI_COMMIT_SHORT_SHA
|
||||||
|
|
||||||
|
docker login -u $CI_REGISTRY_USER -p $CI_REGISTRY_PASSWORD $CI_REGISTRY
|
||||||
|
docker build . -t $IMAGE_TAG -f docker/Dockerfile --build-arg AUTH_PACKAGE=git+https://gitlab.com/allianceauth/allianceauth@$CI_COMMIT_BRANCH
|
||||||
|
docker push $IMAGE_TAG
|
||||||
|
rules:
|
||||||
|
- if: '$CI_MERGE_REQUEST_SOURCE_BRANCH_NAME == ""'
|
||||||
|
when: manual
|
||||||
|
- if: '$CI_MERGE_REQUEST_SOURCE_BRANCH_NAME != ""'
|
||||||
|
when: never
|
||||||
|
|
||||||
|
build-image-mr:
|
||||||
|
before_script: []
|
||||||
|
image: docker:20.10.10
|
||||||
|
stage: docker
|
||||||
|
services:
|
||||||
|
- docker:20.10.10-dind
|
||||||
|
script: |
|
||||||
|
CURRENT_DATE=$(echo $CI_COMMIT_TIMESTAMP | head -c 10 | tr -d -)
|
||||||
|
IMAGE_TAG=$CI_REGISTRY_IMAGE/auth:$CURRENT_DATE-$CI_MERGE_REQUEST_SOURCE_BRANCH_NAME-$CI_COMMIT_SHORT_SHA
|
||||||
|
|
||||||
|
docker login -u $CI_REGISTRY_USER -p $CI_REGISTRY_PASSWORD $CI_REGISTRY
|
||||||
|
docker build . -t $IMAGE_TAG -f docker/Dockerfile --build-arg AUTH_PACKAGE=git+$CI_MERGE_REQUEST_SOURCE_PROJECT_URL@$CI_MERGE_REQUEST_SOURCE_BRANCH_NAME
|
||||||
|
docker push $IMAGE_TAG
|
||||||
|
rules:
|
||||||
|
- if: '$CI_PIPELINE_SOURCE == "merge_request_event"'
|
||||||
|
when: manual
|
||||||
|
- if: '$CI_PIPELINE_SOURCE != "merge_request_event"'
|
||||||
|
when: never
|
||||||
|
|||||||
@@ -22,7 +22,13 @@ repos:
|
|||||||
args: [ '--remove' ]
|
args: [ '--remove' ]
|
||||||
|
|
||||||
- repo: https://github.com/editorconfig-checker/editorconfig-checker.python
|
- repo: https://github.com/editorconfig-checker/editorconfig-checker.python
|
||||||
rev: 2.3.5
|
rev: 2.3.54
|
||||||
hooks:
|
hooks:
|
||||||
- id: editorconfig-checker
|
- id: editorconfig-checker
|
||||||
exclude: ^(LICENSE|allianceauth\/static\/css\/themes\/bootstrap-locals.less|allianceauth\/eveonline\/swagger.json|(.*.po)|(.*.mo))
|
exclude: ^(LICENSE|allianceauth\/static\/css\/themes\/bootstrap-locals.less|allianceauth\/eveonline\/swagger.json|(.*.po)|(.*.mo))
|
||||||
|
|
||||||
|
- repo: https://github.com/asottile/pyupgrade
|
||||||
|
rev: v2.29.0
|
||||||
|
hooks:
|
||||||
|
- id: pyupgrade
|
||||||
|
args: [ --py37-plus ]
|
||||||
|
|||||||
@@ -1,8 +1,8 @@
|
|||||||
# This will make sure the app is always imported when
|
# This will make sure the app is always imported when
|
||||||
# Django starts so that shared_task will use this app.
|
# Django starts so that shared_task will use this app.
|
||||||
|
|
||||||
__version__ = '2.9.0b1'
|
__version__ = '2.11.2'
|
||||||
__title__ = 'Alliance Auth'
|
__title__ = 'Alliance Auth'
|
||||||
__url__ = 'https://gitlab.com/allianceauth/allianceauth'
|
__url__ = 'https://gitlab.com/allianceauth/allianceauth'
|
||||||
NAME = '%s v%s' % (__title__, __version__)
|
NAME = f'{__title__} v{__version__}'
|
||||||
default_app_config = 'allianceauth.apps.AllianceAuthConfig'
|
default_app_config = 'allianceauth.apps.AllianceAuthConfig'
|
||||||
|
|||||||
@@ -1,13 +1,18 @@
|
|||||||
from bs4 import BeautifulSoup
|
from bs4 import BeautifulSoup
|
||||||
|
|
||||||
|
from django.conf import settings
|
||||||
from django.utils.deprecation import MiddlewareMixin
|
from django.utils.deprecation import MiddlewareMixin
|
||||||
from .models import AnalyticsTokens, AnalyticsIdentifier
|
from .models import AnalyticsTokens, AnalyticsIdentifier
|
||||||
from .tasks import send_ga_tracking_web_view
|
from .tasks import send_ga_tracking_web_view
|
||||||
|
|
||||||
|
import re
|
||||||
|
|
||||||
|
|
||||||
class AnalyticsMiddleware(MiddlewareMixin):
|
class AnalyticsMiddleware(MiddlewareMixin):
|
||||||
def process_response(self, request, response):
|
def process_response(self, request, response):
|
||||||
"""Django Middleware: Process Page Views and creates Analytics Celery Tasks"""
|
"""Django Middleware: Process Page Views and creates Analytics Celery Tasks"""
|
||||||
|
if getattr(settings, "ANALYTICS_DISABLED", False):
|
||||||
|
return response
|
||||||
analyticstokens = AnalyticsTokens.objects.all()
|
analyticstokens = AnalyticsTokens.objects.all()
|
||||||
client_id = AnalyticsIdentifier.objects.get(id=1).identifier.hex
|
client_id = AnalyticsIdentifier.objects.get(id=1).identifier.hex
|
||||||
try:
|
try:
|
||||||
@@ -20,7 +25,13 @@ class AnalyticsMiddleware(MiddlewareMixin):
|
|||||||
if token.send_page_views is False:
|
if token.send_page_views is False:
|
||||||
continue
|
continue
|
||||||
# Check Exclusions
|
# Check Exclusions
|
||||||
if request.path in token.ignore_paths.all():
|
ignore = False
|
||||||
|
for ignore_path in token.ignore_paths.values():
|
||||||
|
ignore_path_regex = re.compile(ignore_path["ignore_path"])
|
||||||
|
if re.search(ignore_path_regex, request.path) is not None:
|
||||||
|
ignore = True
|
||||||
|
|
||||||
|
if ignore is True:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
tracking_id = token.token
|
tracking_id = token.token
|
||||||
|
|||||||
42
allianceauth/analytics/migrations/0004_auto_20211015_0502.py
Normal file
42
allianceauth/analytics/migrations/0004_auto_20211015_0502.py
Normal file
@@ -0,0 +1,42 @@
|
|||||||
|
# Generated by Django 3.1.13 on 2021-10-15 05:02
|
||||||
|
|
||||||
|
from django.core.exceptions import ObjectDoesNotExist
|
||||||
|
from django.db import migrations
|
||||||
|
|
||||||
|
|
||||||
|
def modify_aa_team_token_add_page_ignore_paths(apps, schema_editor):
|
||||||
|
# Add /admin/ and /user_notifications_count/ path to ignore
|
||||||
|
|
||||||
|
AnalyticsPath = apps.get_model('analytics', 'AnalyticsPath')
|
||||||
|
admin = AnalyticsPath.objects.create(ignore_path=r"^\/admin\/.*")
|
||||||
|
user_notifications_count = AnalyticsPath.objects.create(ignore_path=r"^\/user_notifications_count\/.*")
|
||||||
|
|
||||||
|
Tokens = apps.get_model('analytics', 'AnalyticsTokens')
|
||||||
|
token = Tokens.objects.get(token="UA-186249766-2")
|
||||||
|
token.ignore_paths.add(admin, user_notifications_count)
|
||||||
|
|
||||||
|
|
||||||
|
def undo_modify_aa_team_token_add_page_ignore_paths(apps, schema_editor):
|
||||||
|
#
|
||||||
|
AnalyticsPath = apps.get_model('analytics', 'AnalyticsPath')
|
||||||
|
Tokens = apps.get_model('analytics', 'AnalyticsTokens')
|
||||||
|
|
||||||
|
token = Tokens.objects.get(token="UA-186249766-2")
|
||||||
|
try:
|
||||||
|
admin = AnalyticsPath.objects.get(ignore_path=r"^\/admin\/.*", analyticstokens=token)
|
||||||
|
user_notifications_count = AnalyticsPath.objects.get(ignore_path=r"^\/user_notifications_count\/.*", analyticstokens=token)
|
||||||
|
admin.delete()
|
||||||
|
user_notifications_count.delete()
|
||||||
|
except ObjectDoesNotExist:
|
||||||
|
# Its fine if it doesnt exist, we just dont want them building up when re-migrating
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('analytics', '0003_Generate_Identifier'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [migrations.RunPython(modify_aa_team_token_add_page_ignore_paths, undo_modify_aa_team_token_add_page_ignore_paths)
|
||||||
|
]
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
# Generated by Django 3.2.8 on 2021-10-17 16:26
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('analytics', '0004_auto_20211015_0502'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='analyticspath',
|
||||||
|
name='ignore_path',
|
||||||
|
field=models.CharField(default='/example/', help_text='Regex Expression, If matched no Analytics Page View is sent', max_length=254),
|
||||||
|
),
|
||||||
|
]
|
||||||
40
allianceauth/analytics/migrations/0006_more_ignore_paths.py
Normal file
40
allianceauth/analytics/migrations/0006_more_ignore_paths.py
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
# Generated by Django 3.2.8 on 2021-10-19 01:47
|
||||||
|
|
||||||
|
from django.core.exceptions import ObjectDoesNotExist
|
||||||
|
from django.db import migrations
|
||||||
|
|
||||||
|
|
||||||
|
def modify_aa_team_token_add_page_ignore_paths(apps, schema_editor):
|
||||||
|
# Add the /account/activate path to ignore
|
||||||
|
|
||||||
|
AnalyticsPath = apps.get_model('analytics', 'AnalyticsPath')
|
||||||
|
account_activate = AnalyticsPath.objects.create(ignore_path=r"^\/account\/activate\/.*")
|
||||||
|
|
||||||
|
Tokens = apps.get_model('analytics', 'AnalyticsTokens')
|
||||||
|
token = Tokens.objects.get(token="UA-186249766-2")
|
||||||
|
token.ignore_paths.add(account_activate)
|
||||||
|
|
||||||
|
|
||||||
|
def undo_modify_aa_team_token_add_page_ignore_paths(apps, schema_editor):
|
||||||
|
#
|
||||||
|
AnalyticsPath = apps.get_model('analytics', 'AnalyticsPath')
|
||||||
|
Tokens = apps.get_model('analytics', 'AnalyticsTokens')
|
||||||
|
|
||||||
|
token = Tokens.objects.get(token="UA-186249766-2")
|
||||||
|
|
||||||
|
try:
|
||||||
|
account_activate = AnalyticsPath.objects.get(ignore_path=r"^\/account\/activate\/.*", analyticstokens=token)
|
||||||
|
account_activate.delete()
|
||||||
|
except ObjectDoesNotExist:
|
||||||
|
# Its fine if it doesnt exist, we just dont want them building up when re-migrating
|
||||||
|
pass
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('analytics', '0005_alter_analyticspath_ignore_path'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.RunPython(modify_aa_team_token_add_page_ignore_paths, undo_modify_aa_team_token_add_page_ignore_paths)
|
||||||
|
]
|
||||||
@@ -16,11 +16,11 @@ class AnalyticsIdentifier(models.Model):
|
|||||||
raise ValidationError('There is can be only one \
|
raise ValidationError('There is can be only one \
|
||||||
AnalyticsIdentifier instance')
|
AnalyticsIdentifier instance')
|
||||||
self.pk = self.id = 1 # If this happens to be deleted and recreated, force it to be 1
|
self.pk = self.id = 1 # If this happens to be deleted and recreated, force it to be 1
|
||||||
return super(AnalyticsIdentifier, self).save(*args, **kwargs)
|
return super().save(*args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
class AnalyticsPath(models.Model):
|
class AnalyticsPath(models.Model):
|
||||||
ignore_path = models.CharField(max_length=254, default="/example/")
|
ignore_path = models.CharField(max_length=254, default="/example/", help_text="Regex Expression, If matched no Analytics Page View is sent")
|
||||||
|
|
||||||
|
|
||||||
class AnalyticsTokens(models.Model):
|
class AnalyticsTokens(models.Model):
|
||||||
|
|||||||
@@ -1,7 +1,8 @@
|
|||||||
from allianceauth.analytics.tasks import analytics_event
|
|
||||||
from celery.signals import task_failure, task_success
|
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
from celery.signals import task_failure, task_success
|
||||||
|
from django.conf import settings
|
||||||
|
from allianceauth.analytics.tasks import analytics_event
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
@@ -11,6 +12,8 @@ def process_failure_signal(
|
|||||||
sender, task_id, signal,
|
sender, task_id, signal,
|
||||||
args, kwargs, einfo, **kw):
|
args, kwargs, einfo, **kw):
|
||||||
logger.debug("Celery task_failure signal %s" % sender.__class__.__name__)
|
logger.debug("Celery task_failure signal %s" % sender.__class__.__name__)
|
||||||
|
if getattr(settings, "ANALYTICS_DISABLED", False):
|
||||||
|
return
|
||||||
|
|
||||||
category = sender.__module__
|
category = sender.__module__
|
||||||
|
|
||||||
@@ -20,7 +23,7 @@ def process_failure_signal(
|
|||||||
|
|
||||||
action = sender.__name__
|
action = sender.__name__
|
||||||
|
|
||||||
label = f"{exception.__class__.__name__}: {str(exception)}"
|
label = f"{exception.__class__.__name__}"
|
||||||
|
|
||||||
analytics_event(category=category,
|
analytics_event(category=category,
|
||||||
action=action,
|
action=action,
|
||||||
@@ -30,6 +33,8 @@ def process_failure_signal(
|
|||||||
@task_success.connect
|
@task_success.connect
|
||||||
def celery_success_signal(sender, result=None, **kw):
|
def celery_success_signal(sender, result=None, **kw):
|
||||||
logger.debug("Celery task_success signal %s" % sender.__class__.__name__)
|
logger.debug("Celery task_success signal %s" % sender.__class__.__name__)
|
||||||
|
if getattr(settings, "ANALYTICS_DISABLED", False):
|
||||||
|
return
|
||||||
|
|
||||||
category = sender.__module__
|
category = sender.__module__
|
||||||
|
|
||||||
|
|||||||
@@ -40,8 +40,7 @@ def analytics_event(category: str,
|
|||||||
Send a Google Analytics Event for each token stored
|
Send a Google Analytics Event for each token stored
|
||||||
Includes check for if its enabled/disabled
|
Includes check for if its enabled/disabled
|
||||||
|
|
||||||
Parameters
|
Args:
|
||||||
-------
|
|
||||||
`category` (str): Celery Namespace
|
`category` (str): Celery Namespace
|
||||||
`action` (str): Task Name
|
`action` (str): Task Name
|
||||||
`label` (str): Optional, Task Success/Exception
|
`label` (str): Optional, Task Success/Exception
|
||||||
@@ -60,20 +59,21 @@ def analytics_event(category: str,
|
|||||||
|
|
||||||
if allowed is True:
|
if allowed is True:
|
||||||
tracking_id = token.token
|
tracking_id = token.token
|
||||||
send_ga_tracking_celery_event.s(tracking_id=tracking_id,
|
send_ga_tracking_celery_event.s(
|
||||||
|
tracking_id=tracking_id,
|
||||||
client_id=client_id,
|
client_id=client_id,
|
||||||
category=category,
|
category=category,
|
||||||
action=action,
|
action=action,
|
||||||
label=label,
|
label=label,
|
||||||
value=value).\
|
value=value).apply_async(priority=9)
|
||||||
apply_async(priority=9)
|
|
||||||
|
|
||||||
|
|
||||||
@shared_task()
|
@shared_task()
|
||||||
def analytics_daily_stats():
|
def analytics_daily_stats():
|
||||||
"""Celery Task: Do not call directly
|
"""Celery Task: Do not call directly
|
||||||
|
|
||||||
Gathers a series of daily statistics and sends analytics events containing them"""
|
Gathers a series of daily statistics and sends analytics events containing them
|
||||||
|
"""
|
||||||
users = install_stat_users()
|
users = install_stat_users()
|
||||||
tokens = install_stat_tokens()
|
tokens = install_stat_tokens()
|
||||||
addons = install_stat_addons()
|
addons = install_stat_addons()
|
||||||
|
|||||||
108
allianceauth/analytics/tests/test_integration.py
Normal file
108
allianceauth/analytics/tests/test_integration.py
Normal file
@@ -0,0 +1,108 @@
|
|||||||
|
from unittest.mock import patch
|
||||||
|
from urllib.parse import parse_qs
|
||||||
|
|
||||||
|
import requests_mock
|
||||||
|
|
||||||
|
from django.test import TestCase, override_settings
|
||||||
|
|
||||||
|
from allianceauth.analytics.tasks import ANALYTICS_URL
|
||||||
|
from allianceauth.eveonline.tasks import update_character
|
||||||
|
from allianceauth.tests.auth_utils import AuthUtils
|
||||||
|
|
||||||
|
|
||||||
|
@override_settings(CELERY_ALWAYS_EAGER=True)
|
||||||
|
@requests_mock.mock()
|
||||||
|
class TestAnalyticsForViews(TestCase):
|
||||||
|
@override_settings(ANALYTICS_DISABLED=False)
|
||||||
|
def test_should_run_analytics(self, requests_mocker):
|
||||||
|
# given
|
||||||
|
requests_mocker.post(ANALYTICS_URL)
|
||||||
|
user = AuthUtils.create_user("Bruce Wayne")
|
||||||
|
self.client.force_login(user)
|
||||||
|
# when
|
||||||
|
response = self.client.get("/dashboard/")
|
||||||
|
# then
|
||||||
|
self.assertEqual(response.status_code, 200)
|
||||||
|
self.assertTrue(requests_mocker.called)
|
||||||
|
|
||||||
|
@override_settings(ANALYTICS_DISABLED=True)
|
||||||
|
def test_should_not_run_analytics(self, requests_mocker):
|
||||||
|
# given
|
||||||
|
requests_mocker.post(ANALYTICS_URL)
|
||||||
|
user = AuthUtils.create_user("Bruce Wayne")
|
||||||
|
self.client.force_login(user)
|
||||||
|
# when
|
||||||
|
response = self.client.get("/dashboard/")
|
||||||
|
# then
|
||||||
|
self.assertEqual(response.status_code, 200)
|
||||||
|
self.assertFalse(requests_mocker.called)
|
||||||
|
|
||||||
|
|
||||||
|
@override_settings(CELERY_ALWAYS_EAGER=True)
|
||||||
|
@requests_mock.mock()
|
||||||
|
class TestAnalyticsForTasks(TestCase):
|
||||||
|
@override_settings(ANALYTICS_DISABLED=False)
|
||||||
|
@patch("allianceauth.eveonline.models.EveCharacter.objects.update_character")
|
||||||
|
def test_should_run_analytics_for_successful_task(
|
||||||
|
self, requests_mocker, mock_update_character
|
||||||
|
):
|
||||||
|
# given
|
||||||
|
requests_mocker.post(ANALYTICS_URL)
|
||||||
|
user = AuthUtils.create_user("Bruce Wayne")
|
||||||
|
character = AuthUtils.add_main_character_2(user, "Bruce Wayne", 1001)
|
||||||
|
# when
|
||||||
|
update_character.delay(character.character_id)
|
||||||
|
# then
|
||||||
|
self.assertTrue(mock_update_character.called)
|
||||||
|
self.assertTrue(requests_mocker.called)
|
||||||
|
payload = parse_qs(requests_mocker.last_request.text)
|
||||||
|
self.assertListEqual(payload["el"], ["Success"])
|
||||||
|
|
||||||
|
@override_settings(ANALYTICS_DISABLED=True)
|
||||||
|
@patch("allianceauth.eveonline.models.EveCharacter.objects.update_character")
|
||||||
|
def test_should_not_run_analytics_for_successful_task(
|
||||||
|
self, requests_mocker, mock_update_character
|
||||||
|
):
|
||||||
|
# given
|
||||||
|
requests_mocker.post(ANALYTICS_URL)
|
||||||
|
user = AuthUtils.create_user("Bruce Wayne")
|
||||||
|
character = AuthUtils.add_main_character_2(user, "Bruce Wayne", 1001)
|
||||||
|
# when
|
||||||
|
update_character.delay(character.character_id)
|
||||||
|
# then
|
||||||
|
self.assertTrue(mock_update_character.called)
|
||||||
|
self.assertFalse(requests_mocker.called)
|
||||||
|
|
||||||
|
@override_settings(ANALYTICS_DISABLED=False)
|
||||||
|
@patch("allianceauth.eveonline.models.EveCharacter.objects.update_character")
|
||||||
|
def test_should_run_analytics_for_failed_task(
|
||||||
|
self, requests_mocker, mock_update_character
|
||||||
|
):
|
||||||
|
# given
|
||||||
|
requests_mocker.post(ANALYTICS_URL)
|
||||||
|
mock_update_character.side_effect = RuntimeError
|
||||||
|
user = AuthUtils.create_user("Bruce Wayne")
|
||||||
|
character = AuthUtils.add_main_character_2(user, "Bruce Wayne", 1001)
|
||||||
|
# when
|
||||||
|
update_character.delay(character.character_id)
|
||||||
|
# then
|
||||||
|
self.assertTrue(mock_update_character.called)
|
||||||
|
self.assertTrue(requests_mocker.called)
|
||||||
|
payload = parse_qs(requests_mocker.last_request.text)
|
||||||
|
self.assertNotEqual(payload["el"], ["Success"])
|
||||||
|
|
||||||
|
@override_settings(ANALYTICS_DISABLED=True)
|
||||||
|
@patch("allianceauth.eveonline.models.EveCharacter.objects.update_character")
|
||||||
|
def test_should_not_run_analytics_for_failed_task(
|
||||||
|
self, requests_mocker, mock_update_character
|
||||||
|
):
|
||||||
|
# given
|
||||||
|
requests_mocker.post(ANALYTICS_URL)
|
||||||
|
mock_update_character.side_effect = RuntimeError
|
||||||
|
user = AuthUtils.create_user("Bruce Wayne")
|
||||||
|
character = AuthUtils.add_main_character_2(user, "Bruce Wayne", 1001)
|
||||||
|
# when
|
||||||
|
update_character.delay(character.character_id)
|
||||||
|
# then
|
||||||
|
self.assertTrue(mock_update_character.called)
|
||||||
|
self.assertFalse(requests_mocker.called)
|
||||||
@@ -17,7 +17,7 @@ from allianceauth.authentication.models import State, get_guest_state,\
|
|||||||
CharacterOwnership, UserProfile, OwnershipRecord
|
CharacterOwnership, UserProfile, OwnershipRecord
|
||||||
from allianceauth.hooks import get_hooks
|
from allianceauth.hooks import get_hooks
|
||||||
from allianceauth.eveonline.models import EveCharacter, EveCorporationInfo,\
|
from allianceauth.eveonline.models import EveCharacter, EveCorporationInfo,\
|
||||||
EveAllianceInfo
|
EveAllianceInfo, EveFactionInfo
|
||||||
from allianceauth.eveonline.tasks import update_character
|
from allianceauth.eveonline.tasks import update_character
|
||||||
from .app_settings import AUTHENTICATION_ADMIN_USERS_MAX_GROUPS, \
|
from .app_settings import AUTHENTICATION_ADMIN_USERS_MAX_GROUPS, \
|
||||||
AUTHENTICATION_ADMIN_USERS_MAX_CHARS
|
AUTHENTICATION_ADMIN_USERS_MAX_CHARS
|
||||||
@@ -36,8 +36,8 @@ def make_service_hooks_update_groups_action(service):
|
|||||||
for user in queryset: # queryset filtering doesn't work here?
|
for user in queryset: # queryset filtering doesn't work here?
|
||||||
service.update_groups(user)
|
service.update_groups(user)
|
||||||
|
|
||||||
update_service_groups.__name__ = str('update_{}_groups'.format(slugify(service.name)))
|
update_service_groups.__name__ = str(f'update_{slugify(service.name)}_groups')
|
||||||
update_service_groups.short_description = "Sync groups for selected {} accounts".format(service.title)
|
update_service_groups.short_description = f"Sync groups for selected {service.title} accounts"
|
||||||
return update_service_groups
|
return update_service_groups
|
||||||
|
|
||||||
|
|
||||||
@@ -54,8 +54,8 @@ def make_service_hooks_sync_nickname_action(service):
|
|||||||
for user in queryset: # queryset filtering doesn't work here?
|
for user in queryset: # queryset filtering doesn't work here?
|
||||||
service.sync_nickname(user)
|
service.sync_nickname(user)
|
||||||
|
|
||||||
sync_nickname.__name__ = str('sync_{}_nickname'.format(slugify(service.name)))
|
sync_nickname.__name__ = str(f'sync_{slugify(service.name)}_nickname')
|
||||||
sync_nickname.short_description = "Sync nicknames for selected {} accounts".format(service.title)
|
sync_nickname.short_description = f"Sync nicknames for selected {service.title} accounts"
|
||||||
return sync_nickname
|
return sync_nickname
|
||||||
|
|
||||||
|
|
||||||
@@ -159,18 +159,14 @@ def user_main_organization(obj):
|
|||||||
"""
|
"""
|
||||||
user_obj = obj.user if hasattr(obj, 'user') else obj
|
user_obj = obj.user if hasattr(obj, 'user') else obj
|
||||||
if not user_obj.profile.main_character:
|
if not user_obj.profile.main_character:
|
||||||
result = None
|
result = ''
|
||||||
else:
|
else:
|
||||||
corporation = user_obj.profile.main_character.corporation_name
|
result = user_obj.profile.main_character.corporation_name
|
||||||
if user_obj.profile.main_character.alliance_id:
|
if user_obj.profile.main_character.alliance_id:
|
||||||
result = format_html(
|
result += f'<br>{user_obj.profile.main_character.alliance_name}'
|
||||||
'{}<br>{}',
|
elif user_obj.profile.main_character.faction_name:
|
||||||
corporation,
|
result += f'<br>{user_obj.profile.main_character.faction_name}'
|
||||||
user_obj.profile.main_character.alliance_name
|
return format_html(result)
|
||||||
)
|
|
||||||
else:
|
|
||||||
result = corporation
|
|
||||||
return result
|
|
||||||
|
|
||||||
|
|
||||||
user_main_organization.short_description = 'Corporation / Alliance (Main)'
|
user_main_organization.short_description = 'Corporation / Alliance (Main)'
|
||||||
@@ -194,7 +190,7 @@ class MainCorporationsFilter(admin.SimpleListFilter):
|
|||||||
.distinct()\
|
.distinct()\
|
||||||
.order_by(Lower('corporation_name'))
|
.order_by(Lower('corporation_name'))
|
||||||
return tuple(
|
return tuple(
|
||||||
[(x['corporation_id'], x['corporation_name']) for x in qs]
|
(x['corporation_id'], x['corporation_name']) for x in qs
|
||||||
)
|
)
|
||||||
|
|
||||||
def queryset(self, request, qs):
|
def queryset(self, request, qs):
|
||||||
@@ -228,7 +224,7 @@ class MainAllianceFilter(admin.SimpleListFilter):
|
|||||||
.distinct()\
|
.distinct()\
|
||||||
.order_by(Lower('alliance_name'))
|
.order_by(Lower('alliance_name'))
|
||||||
return tuple(
|
return tuple(
|
||||||
[(x['alliance_id'], x['alliance_name']) for x in qs]
|
(x['alliance_id'], x['alliance_name']) for x in qs
|
||||||
)
|
)
|
||||||
|
|
||||||
def queryset(self, request, qs):
|
def queryset(self, request, qs):
|
||||||
@@ -243,6 +239,38 @@ class MainAllianceFilter(admin.SimpleListFilter):
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class MainFactionFilter(admin.SimpleListFilter):
|
||||||
|
"""Custom filter to filter on factions from mains only
|
||||||
|
|
||||||
|
works for both User objects and objects with `user` as FK to User
|
||||||
|
To be used for all user based admin lists
|
||||||
|
"""
|
||||||
|
title = 'faction'
|
||||||
|
parameter_name = 'main_faction_id__exact'
|
||||||
|
|
||||||
|
def lookups(self, request, model_admin):
|
||||||
|
qs = EveCharacter.objects\
|
||||||
|
.exclude(faction_id=None)\
|
||||||
|
.exclude(userprofile=None)\
|
||||||
|
.values('faction_id', 'faction_name')\
|
||||||
|
.distinct()\
|
||||||
|
.order_by(Lower('faction_name'))
|
||||||
|
return tuple(
|
||||||
|
(x['faction_id'], x['faction_name']) for x in qs
|
||||||
|
)
|
||||||
|
|
||||||
|
def queryset(self, request, qs):
|
||||||
|
if self.value() is None:
|
||||||
|
return qs.all()
|
||||||
|
else:
|
||||||
|
if qs.model == User:
|
||||||
|
return qs.filter(profile__main_character__faction_id=self.value())
|
||||||
|
else:
|
||||||
|
return qs.filter(
|
||||||
|
user__profile__main_character__faction_id=self.value()
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
def update_main_character_model(modeladmin, request, queryset):
|
def update_main_character_model(modeladmin, request, queryset):
|
||||||
tasks_count = 0
|
tasks_count = 0
|
||||||
for obj in queryset:
|
for obj in queryset:
|
||||||
@@ -252,7 +280,7 @@ def update_main_character_model(modeladmin, request, queryset):
|
|||||||
|
|
||||||
modeladmin.message_user(
|
modeladmin.message_user(
|
||||||
request,
|
request,
|
||||||
'Update from ESI started for {} characters'.format(tasks_count)
|
f'Update from ESI started for {tasks_count} characters'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@@ -342,6 +370,7 @@ class UserAdmin(BaseUserAdmin):
|
|||||||
'groups',
|
'groups',
|
||||||
MainCorporationsFilter,
|
MainCorporationsFilter,
|
||||||
MainAllianceFilter,
|
MainAllianceFilter,
|
||||||
|
MainFactionFilter,
|
||||||
'is_active',
|
'is_active',
|
||||||
'date_joined',
|
'date_joined',
|
||||||
'is_staff',
|
'is_staff',
|
||||||
@@ -351,6 +380,7 @@ class UserAdmin(BaseUserAdmin):
|
|||||||
'username',
|
'username',
|
||||||
'character_ownerships__character__character_name'
|
'character_ownerships__character__character_name'
|
||||||
)
|
)
|
||||||
|
readonly_fields = ('date_joined', 'last_login')
|
||||||
|
|
||||||
def _characters(self, obj):
|
def _characters(self, obj):
|
||||||
character_ownerships = list(obj.character_ownerships.all())
|
character_ownerships = list(obj.character_ownerships.all())
|
||||||
@@ -369,7 +399,7 @@ class UserAdmin(BaseUserAdmin):
|
|||||||
_state.admin_order_field = 'profile__state'
|
_state.admin_order_field = 'profile__state'
|
||||||
|
|
||||||
def _groups(self, obj):
|
def _groups(self, obj):
|
||||||
my_groups = sorted([group.name for group in list(obj.groups.all())])
|
my_groups = sorted(group.name for group in list(obj.groups.all()))
|
||||||
return self._list_2_html_w_tooltips(
|
return self._list_2_html_w_tooltips(
|
||||||
my_groups, AUTHENTICATION_ADMIN_USERS_MAX_GROUPS
|
my_groups, AUTHENTICATION_ADMIN_USERS_MAX_GROUPS
|
||||||
)
|
)
|
||||||
@@ -396,10 +426,19 @@ class UserAdmin(BaseUserAdmin):
|
|||||||
def has_delete_permission(self, request, obj=None):
|
def has_delete_permission(self, request, obj=None):
|
||||||
return request.user.has_perm('auth.delete_user')
|
return request.user.has_perm('auth.delete_user')
|
||||||
|
|
||||||
|
def get_object(self, *args , **kwargs):
|
||||||
|
obj = super().get_object(*args , **kwargs)
|
||||||
|
self.obj = obj # storing current object for use in formfield_for_manytomany
|
||||||
|
return obj
|
||||||
|
|
||||||
def formfield_for_manytomany(self, db_field, request, **kwargs):
|
def formfield_for_manytomany(self, db_field, request, **kwargs):
|
||||||
"""overriding this formfield to have sorted lists in the form"""
|
|
||||||
if db_field.name == "groups":
|
if db_field.name == "groups":
|
||||||
kwargs["queryset"] = Group.objects.all().order_by(Lower('name'))
|
groups_qs = Group.objects.filter(authgroup__states__isnull=True)
|
||||||
|
obj_state = self.obj.profile.state
|
||||||
|
if obj_state:
|
||||||
|
matching_groups_qs = Group.objects.filter(authgroup__states=obj_state)
|
||||||
|
groups_qs = groups_qs | matching_groups_qs
|
||||||
|
kwargs["queryset"] = groups_qs.order_by(Lower('name'))
|
||||||
return super().formfield_for_manytomany(db_field, request, **kwargs)
|
return super().formfield_for_manytomany(db_field, request, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
@@ -426,7 +465,8 @@ class StateAdmin(admin.ModelAdmin):
|
|||||||
'public',
|
'public',
|
||||||
'member_characters',
|
'member_characters',
|
||||||
'member_corporations',
|
'member_corporations',
|
||||||
'member_alliances'
|
'member_alliances',
|
||||||
|
'member_factions'
|
||||||
),
|
),
|
||||||
})
|
})
|
||||||
)
|
)
|
||||||
@@ -434,6 +474,7 @@ class StateAdmin(admin.ModelAdmin):
|
|||||||
'member_characters',
|
'member_characters',
|
||||||
'member_corporations',
|
'member_corporations',
|
||||||
'member_alliances',
|
'member_alliances',
|
||||||
|
'member_factions',
|
||||||
'permissions'
|
'permissions'
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -448,6 +489,9 @@ class StateAdmin(admin.ModelAdmin):
|
|||||||
elif db_field.name == "member_alliances":
|
elif db_field.name == "member_alliances":
|
||||||
kwargs["queryset"] = EveAllianceInfo.objects.all()\
|
kwargs["queryset"] = EveAllianceInfo.objects.all()\
|
||||||
.order_by(Lower('alliance_name'))
|
.order_by(Lower('alliance_name'))
|
||||||
|
elif db_field.name == "member_factions":
|
||||||
|
kwargs["queryset"] = EveFactionInfo.objects.all()\
|
||||||
|
.order_by(Lower('faction_name'))
|
||||||
elif db_field.name == "permissions":
|
elif db_field.name == "permissions":
|
||||||
kwargs["queryset"] = Permission.objects.select_related("content_type").all()
|
kwargs["queryset"] = Permission.objects.select_related("content_type").all()
|
||||||
return super().formfield_for_manytomany(db_field, request, **kwargs)
|
return super().formfield_for_manytomany(db_field, request, **kwargs)
|
||||||
@@ -455,7 +499,7 @@ class StateAdmin(admin.ModelAdmin):
|
|||||||
def has_delete_permission(self, request, obj=None):
|
def has_delete_permission(self, request, obj=None):
|
||||||
if obj == get_guest_state():
|
if obj == get_guest_state():
|
||||||
return False
|
return False
|
||||||
return super(StateAdmin, self).has_delete_permission(request, obj=obj)
|
return super().has_delete_permission(request, obj=obj)
|
||||||
|
|
||||||
def get_fieldsets(self, request, obj=None):
|
def get_fieldsets(self, request, obj=None):
|
||||||
if obj == get_guest_state():
|
if obj == get_guest_state():
|
||||||
@@ -464,7 +508,7 @@ class StateAdmin(admin.ModelAdmin):
|
|||||||
'fields': ('permissions', 'priority'),
|
'fields': ('permissions', 'priority'),
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
return super(StateAdmin, self).get_fieldsets(request, obj=obj)
|
return super().get_fieldsets(request, obj=obj)
|
||||||
|
|
||||||
|
|
||||||
class BaseOwnershipAdmin(admin.ModelAdmin):
|
class BaseOwnershipAdmin(admin.ModelAdmin):
|
||||||
@@ -485,7 +529,8 @@ class BaseOwnershipAdmin(admin.ModelAdmin):
|
|||||||
'user__username',
|
'user__username',
|
||||||
'character__character_name',
|
'character__character_name',
|
||||||
'character__corporation_name',
|
'character__corporation_name',
|
||||||
'character__alliance_name'
|
'character__alliance_name',
|
||||||
|
'character__faction_name'
|
||||||
)
|
)
|
||||||
list_filter = (
|
list_filter = (
|
||||||
MainCorporationsFilter,
|
MainCorporationsFilter,
|
||||||
|
|||||||
@@ -3,10 +3,14 @@ from django.core.checks import register, Tags
|
|||||||
|
|
||||||
|
|
||||||
class AuthenticationConfig(AppConfig):
|
class AuthenticationConfig(AppConfig):
|
||||||
name = 'allianceauth.authentication'
|
name = "allianceauth.authentication"
|
||||||
label = 'authentication'
|
label = "authentication"
|
||||||
|
|
||||||
def ready(self):
|
def ready(self):
|
||||||
super(AuthenticationConfig, self).ready()
|
from allianceauth.authentication import checks, signals # noqa: F401
|
||||||
from allianceauth.authentication import checks, signals
|
from allianceauth.authentication.task_statistics import (
|
||||||
|
signals as celery_signals,
|
||||||
|
)
|
||||||
|
|
||||||
register(Tags.security)(checks.check_login_scopes_setting)
|
register(Tags.security)(checks.check_login_scopes_setting)
|
||||||
|
celery_signals.reset_counters()
|
||||||
|
|||||||
@@ -36,17 +36,17 @@ class StateBackend(ModelBackend):
|
|||||||
try:
|
try:
|
||||||
ownership = CharacterOwnership.objects.get(character__character_id=token.character_id)
|
ownership = CharacterOwnership.objects.get(character__character_id=token.character_id)
|
||||||
if ownership.owner_hash == token.character_owner_hash:
|
if ownership.owner_hash == token.character_owner_hash:
|
||||||
logger.debug('Authenticating {0} by ownership of character {1}'.format(ownership.user, token.character_name))
|
logger.debug(f'Authenticating {ownership.user} by ownership of character {token.character_name}')
|
||||||
return ownership.user
|
return ownership.user
|
||||||
else:
|
else:
|
||||||
logger.debug('{0} has changed ownership. Creating new user account.'.format(token.character_name))
|
logger.debug(f'{token.character_name} has changed ownership. Creating new user account.')
|
||||||
ownership.delete()
|
ownership.delete()
|
||||||
return self.create_user(token)
|
return self.create_user(token)
|
||||||
except CharacterOwnership.DoesNotExist:
|
except CharacterOwnership.DoesNotExist:
|
||||||
try:
|
try:
|
||||||
# insecure legacy main check for pre-sso registration auth installs
|
# insecure legacy main check for pre-sso registration auth installs
|
||||||
profile = UserProfile.objects.get(main_character__character_id=token.character_id)
|
profile = UserProfile.objects.get(main_character__character_id=token.character_id)
|
||||||
logger.debug('Authenticating {0} by their main character {1} without active ownership.'.format(profile.user, profile.main_character))
|
logger.debug(f'Authenticating {profile.user} by their main character {profile.main_character} without active ownership.')
|
||||||
# attach an ownership
|
# attach an ownership
|
||||||
token.user = profile.user
|
token.user = profile.user
|
||||||
CharacterOwnership.objects.create_by_token(token)
|
CharacterOwnership.objects.create_by_token(token)
|
||||||
@@ -59,13 +59,13 @@ class StateBackend(ModelBackend):
|
|||||||
user = records[0].user
|
user = records[0].user
|
||||||
token.user = user
|
token.user = user
|
||||||
co = CharacterOwnership.objects.create_by_token(token)
|
co = CharacterOwnership.objects.create_by_token(token)
|
||||||
logger.debug('Authenticating {0} by matching owner hash record of character {1}'.format(user, co.character))
|
logger.debug(f'Authenticating {user} by matching owner hash record of character {co.character}')
|
||||||
if not user.profile.main_character:
|
if not user.profile.main_character:
|
||||||
# set this as their main by default if they have none
|
# set this as their main by default if they have none
|
||||||
user.profile.main_character = co.character
|
user.profile.main_character = co.character
|
||||||
user.profile.save()
|
user.profile.save()
|
||||||
return user
|
return user
|
||||||
logger.debug('Unable to authenticate character {0}. Creating new user.'.format(token.character_name))
|
logger.debug(f'Unable to authenticate character {token.character_name}. Creating new user.')
|
||||||
return self.create_user(token)
|
return self.create_user(token)
|
||||||
|
|
||||||
def create_user(self, token):
|
def create_user(self, token):
|
||||||
@@ -77,7 +77,7 @@ class StateBackend(ModelBackend):
|
|||||||
co = CharacterOwnership.objects.create_by_token(token) # assign ownership to this user
|
co = CharacterOwnership.objects.create_by_token(token) # assign ownership to this user
|
||||||
user.profile.main_character = co.character # assign main character as token character
|
user.profile.main_character = co.character # assign main character as token character
|
||||||
user.profile.save()
|
user.profile.save()
|
||||||
logger.debug('Created new user {0}'.format(user))
|
logger.debug(f'Created new user {user}')
|
||||||
return user
|
return user
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
@@ -87,10 +87,10 @@ class StateBackend(ModelBackend):
|
|||||||
if User.objects.filter(username__startswith=name).exists():
|
if User.objects.filter(username__startswith=name).exists():
|
||||||
u = User.objects.filter(username__startswith=name)
|
u = User.objects.filter(username__startswith=name)
|
||||||
num = len(u)
|
num = len(u)
|
||||||
username = "%s_%s" % (name, num)
|
username = f"{name}_{num}"
|
||||||
while u.filter(username=username).exists():
|
while u.filter(username=username).exists():
|
||||||
num += 1
|
num += 1
|
||||||
username = "%s_%s" % (name, num)
|
username = f"{name}_{num}"
|
||||||
else:
|
else:
|
||||||
username = name
|
username = name
|
||||||
return username
|
return username
|
||||||
|
|||||||
@@ -11,10 +11,10 @@ class Command(BaseCommand):
|
|||||||
if profiles.exists():
|
if profiles.exists():
|
||||||
for profile in profiles:
|
for profile in profiles:
|
||||||
self.stdout.write(self.style.ERROR(
|
self.stdout.write(self.style.ERROR(
|
||||||
'{0} does not have an ownership. Resetting user {1} main character.'.format(profile.main_character,
|
'{} does not have an ownership. Resetting user {} main character.'.format(profile.main_character,
|
||||||
profile.user)))
|
profile.user)))
|
||||||
profile.main_character = None
|
profile.main_character = None
|
||||||
profile.save()
|
profile.save()
|
||||||
self.stdout.write(self.style.WARNING('Reset {0} main characters.'.format(profiles.count())))
|
self.stdout.write(self.style.WARNING(f'Reset {profiles.count()} main characters.'))
|
||||||
else:
|
else:
|
||||||
self.stdout.write(self.style.SUCCESS('All main characters have active ownership.'))
|
self.stdout.write(self.style.SUCCESS('All main characters have active ownership.'))
|
||||||
|
|||||||
@@ -16,6 +16,8 @@ def available_states_query(character):
|
|||||||
query |= Q(member_corporations__corporation_id=character.corporation_id)
|
query |= Q(member_corporations__corporation_id=character.corporation_id)
|
||||||
if character.alliance_id:
|
if character.alliance_id:
|
||||||
query |= Q(member_alliances__alliance_id=character.alliance_id)
|
query |= Q(member_alliances__alliance_id=character.alliance_id)
|
||||||
|
if character.faction_id:
|
||||||
|
query |= Q(member_factions__faction_id=character.faction_id)
|
||||||
return query
|
return query
|
||||||
|
|
||||||
|
|
||||||
@@ -49,7 +51,7 @@ class StateQuerySet(QuerySet):
|
|||||||
for state in self:
|
for state in self:
|
||||||
for profile in state.userprofile_set.all():
|
for profile in state.userprofile_set.all():
|
||||||
profile.assign_state(state=self.model.objects.exclude(pk=state.pk).get_for_user(profile.user))
|
profile.assign_state(state=self.model.objects.exclude(pk=state.pk).get_for_user(profile.user))
|
||||||
super(StateQuerySet, self).delete()
|
super().delete()
|
||||||
|
|
||||||
|
|
||||||
class StateManager(Manager):
|
class StateManager(Manager):
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-05 21:38
|
# Generated by Django 1.10.1 on 2016-09-05 21:38
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-07 19:14
|
# Generated by Django 1.10.1 on 2016-09-07 19:14
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-09 20:29
|
# Generated by Django 1.10.1 on 2016-09-09 20:29
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-09 23:19
|
# Generated by Django 1.10.1 on 2016-09-09 23:19
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-09 23:11
|
# Generated by Django 1.10.1 on 2016-09-09 23:11
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-10 05:42
|
# Generated by Django 1.10.1 on 2016-09-10 05:42
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-10 21:50
|
# Generated by Django 1.10.1 on 2016-09-10 21:50
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-12 13:04
|
# Generated by Django 1.10.1 on 2016-09-12 13:04
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.2 on 2016-10-21 02:28
|
# Generated by Django 1.10.2 on 2016-10-21 02:28
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2017-01-07 06:47
|
# Generated by Django 1.10.1 on 2017-01-07 06:47
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
@@ -9,7 +8,7 @@ def count_completed_fields(model):
|
|||||||
def forward(apps, schema_editor):
|
def forward(apps, schema_editor):
|
||||||
# this ensures only one model exists per user
|
# this ensures only one model exists per user
|
||||||
AuthServicesInfo = apps.get_model('authentication', 'AuthServicesInfo')
|
AuthServicesInfo = apps.get_model('authentication', 'AuthServicesInfo')
|
||||||
users = set([a.user for a in AuthServicesInfo.objects.all()])
|
users = {a.user for a in AuthServicesInfo.objects.all()}
|
||||||
for u in users:
|
for u in users:
|
||||||
auths = AuthServicesInfo.objects.filter(user=u)
|
auths = AuthServicesInfo.objects.filter(user=u)
|
||||||
if auths.count() > 1:
|
if auths.count() > 1:
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2017-01-07 07:11
|
# Generated by Django 1.10.1 on 2017-01-07 07:11
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.5 on 2017-01-12 00:59
|
# Generated by Django 1.10.5 on 2017-01-12 00:59
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.2 on 2016-12-11 23:14
|
# Generated by Django 1.10.2 on 2016-12-11 23:14
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-09 23:19
|
# Generated by Django 1.10.1 on 2016-09-09 23:19
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.5 on 2017-03-22 23:09
|
# Generated by Django 1.10.5 on 2017-03-22 23:09
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
import allianceauth.authentication.models
|
import allianceauth.authentication.models
|
||||||
import django.db.models.deletion
|
import django.db.models.deletion
|
||||||
|
|||||||
@@ -1,5 +1,3 @@
|
|||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,18 @@
|
|||||||
|
# Generated by Django 3.2.8 on 2021-10-20 05:22
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('authentication', '0017_remove_fleetup_permission'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AlterField(
|
||||||
|
model_name='state',
|
||||||
|
name='name',
|
||||||
|
field=models.CharField(max_length=32, unique=True),
|
||||||
|
),
|
||||||
|
]
|
||||||
@@ -0,0 +1,19 @@
|
|||||||
|
# Generated by Django 3.1.13 on 2021-10-12 20:21
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('eveonline', '0015_factions'),
|
||||||
|
('authentication', '0017_remove_fleetup_permission'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='state',
|
||||||
|
name='member_factions',
|
||||||
|
field=models.ManyToManyField(blank=True, help_text='Factions to whose members this state is available.', to='eveonline.EveFactionInfo'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@@ -0,0 +1,14 @@
|
|||||||
|
# Generated by Django 3.2.8 on 2021-10-26 09:19
|
||||||
|
|
||||||
|
from django.db import migrations
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('authentication', '0018_alter_state_name_length'),
|
||||||
|
('authentication', '0018_state_member_factions'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
]
|
||||||
@@ -3,7 +3,7 @@ import logging
|
|||||||
from django.contrib.auth.models import User, Permission
|
from django.contrib.auth.models import User, Permission
|
||||||
from django.db import models, transaction
|
from django.db import models, transaction
|
||||||
from django.utils.translation import ugettext_lazy as _
|
from django.utils.translation import ugettext_lazy as _
|
||||||
from allianceauth.eveonline.models import EveCharacter, EveCorporationInfo, EveAllianceInfo
|
from allianceauth.eveonline.models import EveCharacter, EveCorporationInfo, EveAllianceInfo, EveFactionInfo
|
||||||
from allianceauth.notifications import notify
|
from allianceauth.notifications import notify
|
||||||
|
|
||||||
from .managers import CharacterOwnershipManager, StateManager
|
from .managers import CharacterOwnershipManager, StateManager
|
||||||
@@ -12,13 +12,18 @@ logger = logging.getLogger(__name__)
|
|||||||
|
|
||||||
|
|
||||||
class State(models.Model):
|
class State(models.Model):
|
||||||
name = models.CharField(max_length=20, unique=True)
|
name = models.CharField(max_length=32, unique=True)
|
||||||
permissions = models.ManyToManyField(Permission, blank=True)
|
permissions = models.ManyToManyField(Permission, blank=True)
|
||||||
priority = models.IntegerField(unique=True, help_text="Users get assigned the state with the highest priority available to them.")
|
priority = models.IntegerField(unique=True, help_text="Users get assigned the state with the highest priority available to them.")
|
||||||
|
|
||||||
member_characters = models.ManyToManyField(EveCharacter, blank=True, help_text="Characters to which this state is available.")
|
member_characters = models.ManyToManyField(EveCharacter, blank=True,
|
||||||
member_corporations = models.ManyToManyField(EveCorporationInfo, blank=True, help_text="Corporations to whose members this state is available.")
|
help_text="Characters to which this state is available.")
|
||||||
member_alliances = models.ManyToManyField(EveAllianceInfo, blank=True, help_text="Alliances to whose members this state is available.")
|
member_corporations = models.ManyToManyField(EveCorporationInfo, blank=True,
|
||||||
|
help_text="Corporations to whose members this state is available.")
|
||||||
|
member_alliances = models.ManyToManyField(EveAllianceInfo, blank=True,
|
||||||
|
help_text="Alliances to whose members this state is available.")
|
||||||
|
member_factions = models.ManyToManyField(EveFactionInfo, blank=True,
|
||||||
|
help_text="Factions to whose members this state is available.")
|
||||||
public = models.BooleanField(default=False, help_text="Make this state available to any character.")
|
public = models.BooleanField(default=False, help_text="Make this state available to any character.")
|
||||||
|
|
||||||
objects = StateManager()
|
objects = StateManager()
|
||||||
@@ -39,7 +44,7 @@ class State(models.Model):
|
|||||||
with transaction.atomic():
|
with transaction.atomic():
|
||||||
for profile in self.userprofile_set.all():
|
for profile in self.userprofile_set.all():
|
||||||
profile.assign_state(state=State.objects.exclude(pk=self.pk).get_for_user(profile.user))
|
profile.assign_state(state=State.objects.exclude(pk=self.pk).get_for_user(profile.user))
|
||||||
super(State, self).delete(**kwargs)
|
super().delete(**kwargs)
|
||||||
|
|
||||||
|
|
||||||
def get_guest_state():
|
def get_guest_state():
|
||||||
@@ -67,7 +72,7 @@ class UserProfile(models.Model):
|
|||||||
if self.state != state:
|
if self.state != state:
|
||||||
self.state = state
|
self.state = state
|
||||||
if commit:
|
if commit:
|
||||||
logger.info('Updating {} state to {}'.format(self.user, self.state))
|
logger.info(f'Updating {self.user} state to {self.state}')
|
||||||
self.save(update_fields=['state'])
|
self.save(update_fields=['state'])
|
||||||
notify(
|
notify(
|
||||||
self.user,
|
self.user,
|
||||||
@@ -102,7 +107,7 @@ class CharacterOwnership(models.Model):
|
|||||||
objects = CharacterOwnershipManager()
|
objects = CharacterOwnershipManager()
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return "%s: %s" % (self.user, self.character)
|
return f"{self.user}: {self.character}"
|
||||||
|
|
||||||
|
|
||||||
class OwnershipRecord(models.Model):
|
class OwnershipRecord(models.Model):
|
||||||
@@ -115,4 +120,4 @@ class OwnershipRecord(models.Model):
|
|||||||
ordering = ['-created']
|
ordering = ['-created']
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return "%s: %s on %s" % (self.user, self.character, self.created)
|
return f"{self.user}: {self.character} on {self.created}"
|
||||||
|
|||||||
@@ -29,27 +29,32 @@ def trigger_state_check(state):
|
|||||||
@receiver(m2m_changed, sender=State.member_characters.through)
|
@receiver(m2m_changed, sender=State.member_characters.through)
|
||||||
def state_member_characters_changed(sender, instance, action, *args, **kwargs):
|
def state_member_characters_changed(sender, instance, action, *args, **kwargs):
|
||||||
if action.startswith('post_'):
|
if action.startswith('post_'):
|
||||||
logger.debug('State {} member characters changed. Re-evaluating membership.'.format(instance))
|
logger.debug(f'State {instance} member characters changed. Re-evaluating membership.')
|
||||||
trigger_state_check(instance)
|
trigger_state_check(instance)
|
||||||
|
|
||||||
|
|
||||||
@receiver(m2m_changed, sender=State.member_corporations.through)
|
@receiver(m2m_changed, sender=State.member_corporations.through)
|
||||||
def state_member_corporations_changed(sender, instance, action, *args, **kwargs):
|
def state_member_corporations_changed(sender, instance, action, *args, **kwargs):
|
||||||
if action.startswith('post_'):
|
if action.startswith('post_'):
|
||||||
logger.debug('State {} member corporations changed. Re-evaluating membership.'.format(instance))
|
logger.debug(f'State {instance} member corporations changed. Re-evaluating membership.')
|
||||||
trigger_state_check(instance)
|
trigger_state_check(instance)
|
||||||
|
|
||||||
|
|
||||||
@receiver(m2m_changed, sender=State.member_alliances.through)
|
@receiver(m2m_changed, sender=State.member_alliances.through)
|
||||||
def state_member_alliances_changed(sender, instance, action, *args, **kwargs):
|
def state_member_alliances_changed(sender, instance, action, *args, **kwargs):
|
||||||
if action.startswith('post_'):
|
if action.startswith('post_'):
|
||||||
logger.debug('State {} member alliances changed. Re-evaluating membership.'.format(instance))
|
logger.debug(f'State {instance} member alliances changed. Re-evaluating membership.')
|
||||||
trigger_state_check(instance)
|
trigger_state_check(instance)
|
||||||
|
|
||||||
|
@receiver(m2m_changed, sender=State.member_factions.through)
|
||||||
|
def state_member_factions_changed(sender, instance, action, *args, **kwargs):
|
||||||
|
if action.startswith('post_'):
|
||||||
|
logger.debug(f'State {instance} member factions changed. Re-evaluating membership.')
|
||||||
|
trigger_state_check(instance)
|
||||||
|
|
||||||
@receiver(post_save, sender=State)
|
@receiver(post_save, sender=State)
|
||||||
def state_saved(sender, instance, *args, **kwargs):
|
def state_saved(sender, instance, *args, **kwargs):
|
||||||
logger.debug('State {} saved. Re-evaluating membership.'.format(instance))
|
logger.debug(f'State {instance} saved. Re-evaluating membership.')
|
||||||
trigger_state_check(instance)
|
trigger_state_check(instance)
|
||||||
|
|
||||||
|
|
||||||
@@ -60,7 +65,7 @@ def reassess_on_profile_save(sender, instance, created, *args, **kwargs):
|
|||||||
if not created:
|
if not created:
|
||||||
update_fields = kwargs.pop('update_fields', []) or []
|
update_fields = kwargs.pop('update_fields', []) or []
|
||||||
if 'state' not in update_fields:
|
if 'state' not in update_fields:
|
||||||
logger.debug('Profile for {} saved without state change. Re-evaluating state.'.format(instance.user))
|
logger.debug(f'Profile for {instance.user} saved without state change. Re-evaluating state.')
|
||||||
instance.assign_state()
|
instance.assign_state()
|
||||||
|
|
||||||
|
|
||||||
@@ -68,14 +73,14 @@ def reassess_on_profile_save(sender, instance, created, *args, **kwargs):
|
|||||||
def create_required_models(sender, instance, created, *args, **kwargs):
|
def create_required_models(sender, instance, created, *args, **kwargs):
|
||||||
# ensure all users have a model
|
# ensure all users have a model
|
||||||
if created:
|
if created:
|
||||||
logger.debug('User {} created. Creating default UserProfile.'.format(instance))
|
logger.debug(f'User {instance} created. Creating default UserProfile.')
|
||||||
UserProfile.objects.get_or_create(user=instance)
|
UserProfile.objects.get_or_create(user=instance)
|
||||||
|
|
||||||
|
|
||||||
@receiver(post_save, sender=Token)
|
@receiver(post_save, sender=Token)
|
||||||
def record_character_ownership(sender, instance, created, *args, **kwargs):
|
def record_character_ownership(sender, instance, created, *args, **kwargs):
|
||||||
if created:
|
if created:
|
||||||
logger.debug('New token for {0} character {1} saved. Evaluating ownership.'.format(instance.user, instance.character_name))
|
logger.debug(f'New token for {instance.user} character {instance.character_name} saved. Evaluating ownership.')
|
||||||
if instance.user:
|
if instance.user:
|
||||||
query = Q(owner_hash=instance.character_owner_hash) & Q(user=instance.user)
|
query = Q(owner_hash=instance.character_owner_hash) & Q(user=instance.user)
|
||||||
else:
|
else:
|
||||||
@@ -84,13 +89,13 @@ def record_character_ownership(sender, instance, created, *args, **kwargs):
|
|||||||
CharacterOwnership.objects.filter(character__character_id=instance.character_id).exclude(query).delete()
|
CharacterOwnership.objects.filter(character__character_id=instance.character_id).exclude(query).delete()
|
||||||
# create character if needed
|
# create character if needed
|
||||||
if EveCharacter.objects.filter(character_id=instance.character_id).exists() is False:
|
if EveCharacter.objects.filter(character_id=instance.character_id).exists() is False:
|
||||||
logger.debug('Token is for a new character. Creating model for {0} ({1})'.format(instance.character_name, instance.character_id))
|
logger.debug(f'Token is for a new character. Creating model for {instance.character_name} ({instance.character_id})')
|
||||||
EveCharacter.objects.create_character(instance.character_id)
|
EveCharacter.objects.create_character(instance.character_id)
|
||||||
char = EveCharacter.objects.get(character_id=instance.character_id)
|
char = EveCharacter.objects.get(character_id=instance.character_id)
|
||||||
# check if we need to create ownership
|
# check if we need to create ownership
|
||||||
if instance.user and not CharacterOwnership.objects.filter(
|
if instance.user and not CharacterOwnership.objects.filter(
|
||||||
character__character_id=instance.character_id).exists():
|
character__character_id=instance.character_id).exists():
|
||||||
logger.debug("Character {0} is not yet owned. Assigning ownership to {1}".format(instance.character_name, instance.user))
|
logger.debug(f"Character {instance.character_name} is not yet owned. Assigning ownership to {instance.user}")
|
||||||
CharacterOwnership.objects.update_or_create(character=char, defaults={'owner_hash': instance.character_owner_hash, 'user': instance.user})
|
CharacterOwnership.objects.update_or_create(character=char, defaults={'owner_hash': instance.character_owner_hash, 'user': instance.user})
|
||||||
|
|
||||||
|
|
||||||
@@ -98,7 +103,7 @@ def record_character_ownership(sender, instance, created, *args, **kwargs):
|
|||||||
def validate_main_character(sender, instance, *args, **kwargs):
|
def validate_main_character(sender, instance, *args, **kwargs):
|
||||||
try:
|
try:
|
||||||
if instance.user.profile.main_character == instance.character:
|
if instance.user.profile.main_character == instance.character:
|
||||||
logger.info("Ownership of a main character {0} has been revoked. Resetting {1} main character.".format(
|
logger.info("Ownership of a main character {} has been revoked. Resetting {} main character.".format(
|
||||||
instance.character, instance.user))
|
instance.character, instance.user))
|
||||||
# clear main character as user no longer owns them
|
# clear main character as user no longer owns them
|
||||||
instance.user.profile.main_character = None
|
instance.user.profile.main_character = None
|
||||||
@@ -111,7 +116,7 @@ def validate_main_character(sender, instance, *args, **kwargs):
|
|||||||
@receiver(post_delete, sender=Token)
|
@receiver(post_delete, sender=Token)
|
||||||
def validate_ownership(sender, instance, *args, **kwargs):
|
def validate_ownership(sender, instance, *args, **kwargs):
|
||||||
if not Token.objects.filter(character_owner_hash=instance.character_owner_hash).filter(refresh_token__isnull=False).exists():
|
if not Token.objects.filter(character_owner_hash=instance.character_owner_hash).filter(refresh_token__isnull=False).exists():
|
||||||
logger.info("No remaining tokens to validate ownership of character {0}. Revoking ownership.".format(instance.character_name))
|
logger.info(f"No remaining tokens to validate ownership of character {instance.character_name}. Revoking ownership.")
|
||||||
CharacterOwnership.objects.filter(owner_hash=instance.character_owner_hash).delete()
|
CharacterOwnership.objects.filter(owner_hash=instance.character_owner_hash).delete()
|
||||||
|
|
||||||
|
|
||||||
@@ -122,11 +127,11 @@ def assign_state_on_active_change(sender, instance, *args, **kwargs):
|
|||||||
old_instance = User.objects.get(pk=instance.pk)
|
old_instance = User.objects.get(pk=instance.pk)
|
||||||
if old_instance.is_active != instance.is_active:
|
if old_instance.is_active != instance.is_active:
|
||||||
if instance.is_active:
|
if instance.is_active:
|
||||||
logger.debug("User {0} has been activated. Assigning state.".format(instance))
|
logger.debug(f"User {instance} has been activated. Assigning state.")
|
||||||
instance.profile.assign_state()
|
instance.profile.assign_state()
|
||||||
else:
|
else:
|
||||||
logger.debug(
|
logger.debug(
|
||||||
"User {0} has been deactivated. Revoking state and assigning to guest state.".format(instance))
|
f"User {instance} has been deactivated. Revoking state and assigning to guest state.")
|
||||||
instance.profile.state = get_guest_state()
|
instance.profile.state = get_guest_state()
|
||||||
instance.profile.save(update_fields=['state'])
|
instance.profile.save(update_fields=['state'])
|
||||||
|
|
||||||
@@ -135,10 +140,10 @@ def assign_state_on_active_change(sender, instance, *args, **kwargs):
|
|||||||
def check_state_on_character_update(sender, instance, *args, **kwargs):
|
def check_state_on_character_update(sender, instance, *args, **kwargs):
|
||||||
# if this is a main character updating, check that user's state
|
# if this is a main character updating, check that user's state
|
||||||
try:
|
try:
|
||||||
logger.debug("Character {0} has been saved. Assessing owner's state for changes.".format(instance))
|
logger.debug(f"Character {instance} has been saved. Assessing owner's state for changes.")
|
||||||
instance.userprofile.assign_state()
|
instance.userprofile.assign_state()
|
||||||
except UserProfile.DoesNotExist:
|
except UserProfile.DoesNotExist:
|
||||||
logger.debug("Character {0} is not a main character. No state assessment required.".format(instance))
|
logger.debug(f"Character {instance} is not a main character. No state assessment required.")
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
@@ -148,7 +153,7 @@ def ownership_record_creation(sender, instance, created, *args, **kwargs):
|
|||||||
records = OwnershipRecord.objects.filter(owner_hash=instance.owner_hash).filter(character=instance.character)
|
records = OwnershipRecord.objects.filter(owner_hash=instance.owner_hash).filter(character=instance.character)
|
||||||
if records.exists():
|
if records.exists():
|
||||||
if records[0].user == instance.user: # most recent record is sorted first
|
if records[0].user == instance.user: # most recent record is sorted first
|
||||||
logger.debug("Already have ownership record of {0} by user {1}".format(instance.character, instance.user))
|
logger.debug(f"Already have ownership record of {instance.character} by user {instance.user}")
|
||||||
return
|
return
|
||||||
logger.info("Character {0} has a new owner {1}. Creating ownership record.".format(instance.character, instance.user))
|
logger.info(f"Character {instance.character} has a new owner {instance.user}. Creating ownership record.")
|
||||||
OwnershipRecord.objects.create(user=instance.user, character=instance.character, owner_hash=instance.owner_hash)
|
OwnershipRecord.objects.create(user=instance.user, character=instance.character, owner_hash=instance.owner_hash)
|
||||||
|
|||||||
40
allianceauth/authentication/task_statistics/counters.py
Normal file
40
allianceauth/authentication/task_statistics/counters.py
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
from collections import namedtuple
|
||||||
|
import datetime as dt
|
||||||
|
|
||||||
|
from .event_series import EventSeries
|
||||||
|
|
||||||
|
|
||||||
|
"""Global series for counting task events."""
|
||||||
|
succeeded_tasks = EventSeries("SUCCEEDED_TASKS")
|
||||||
|
retried_tasks = EventSeries("RETRIED_TASKS")
|
||||||
|
failed_tasks = EventSeries("FAILED_TASKS")
|
||||||
|
|
||||||
|
|
||||||
|
_TaskCounts = namedtuple(
|
||||||
|
"_TaskCounts", ["succeeded", "retried", "failed", "total", "earliest_task", "hours"]
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def dashboard_results(hours: int) -> _TaskCounts:
|
||||||
|
"""Counts of all task events within the given timeframe."""
|
||||||
|
|
||||||
|
def earliest_if_exists(events: EventSeries, earliest: dt.datetime) -> list:
|
||||||
|
my_earliest = events.first_event(earliest=earliest)
|
||||||
|
return [my_earliest] if my_earliest else []
|
||||||
|
|
||||||
|
earliest = dt.datetime.utcnow() - dt.timedelta(hours=hours)
|
||||||
|
earliest_events = list()
|
||||||
|
succeeded_count = succeeded_tasks.count(earliest=earliest)
|
||||||
|
earliest_events += earliest_if_exists(succeeded_tasks, earliest)
|
||||||
|
retried_count = retried_tasks.count(earliest=earliest)
|
||||||
|
earliest_events += earliest_if_exists(retried_tasks, earliest)
|
||||||
|
failed_count = failed_tasks.count(earliest=earliest)
|
||||||
|
earliest_events += earliest_if_exists(failed_tasks, earliest)
|
||||||
|
return _TaskCounts(
|
||||||
|
succeeded=succeeded_count,
|
||||||
|
retried=retried_count,
|
||||||
|
failed=failed_count,
|
||||||
|
total=succeeded_count + retried_count + failed_count,
|
||||||
|
earliest_task=min(earliest_events) if earliest_events else None,
|
||||||
|
hours=hours,
|
||||||
|
)
|
||||||
95
allianceauth/authentication/task_statistics/event_series.py
Normal file
95
allianceauth/authentication/task_statistics/event_series.py
Normal file
@@ -0,0 +1,95 @@
|
|||||||
|
import datetime as dt
|
||||||
|
from typing import Optional, List
|
||||||
|
|
||||||
|
from redis import Redis
|
||||||
|
from pytz import utc
|
||||||
|
|
||||||
|
from django.core.cache import cache
|
||||||
|
|
||||||
|
|
||||||
|
class EventSeries:
|
||||||
|
"""API for recording and analysing a series of events."""
|
||||||
|
|
||||||
|
_ROOT_KEY = "ALLIANCEAUTH_EVENT_SERIES"
|
||||||
|
|
||||||
|
def __init__(self, key_id: str, redis: Redis = None) -> None:
|
||||||
|
self._redis = cache.get_master_client() if not redis else redis
|
||||||
|
if not isinstance(self._redis, Redis):
|
||||||
|
raise TypeError(
|
||||||
|
"This class requires a Redis client, but none was provided "
|
||||||
|
"and the default Django cache backend is not Redis either."
|
||||||
|
)
|
||||||
|
self._key_id = str(key_id)
|
||||||
|
self.clear()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _key_counter(self):
|
||||||
|
return f"{self._ROOT_KEY}_{self._key_id}_COUNTER"
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _key_sorted_set(self):
|
||||||
|
return f"{self._ROOT_KEY}_{self._key_id}_SORTED_SET"
|
||||||
|
|
||||||
|
def add(self, event_time: dt.datetime = None) -> None:
|
||||||
|
"""Add event.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
- event_time: timestamp of event. Will use current time if not specified.
|
||||||
|
"""
|
||||||
|
if not event_time:
|
||||||
|
event_time = dt.datetime.utcnow()
|
||||||
|
id = self._redis.incr(self._key_counter)
|
||||||
|
self._redis.zadd(self._key_sorted_set, {id: event_time.timestamp()})
|
||||||
|
|
||||||
|
def all(self) -> List[dt.datetime]:
|
||||||
|
"""List of all known events."""
|
||||||
|
return [
|
||||||
|
event[1]
|
||||||
|
for event in self._redis.zrangebyscore(
|
||||||
|
self._key_sorted_set,
|
||||||
|
"-inf",
|
||||||
|
"+inf",
|
||||||
|
withscores=True,
|
||||||
|
score_cast_func=self._cast_scores_to_dt,
|
||||||
|
)
|
||||||
|
]
|
||||||
|
|
||||||
|
def clear(self) -> None:
|
||||||
|
"""Clear all events."""
|
||||||
|
self._redis.delete(self._key_sorted_set)
|
||||||
|
self._redis.delete(self._key_counter)
|
||||||
|
|
||||||
|
def count(self, earliest: dt.datetime = None, latest: dt.datetime = None) -> int:
|
||||||
|
"""Count of events, can be restricted to given timeframe.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
- earliest: Date of first events to count(inclusive), or -infinite if not specified
|
||||||
|
- latest: Date of last events to count(inclusive), or +infinite if not specified
|
||||||
|
"""
|
||||||
|
min = "-inf" if not earliest else earliest.timestamp()
|
||||||
|
max = "+inf" if not latest else latest.timestamp()
|
||||||
|
return self._redis.zcount(self._key_sorted_set, min=min, max=max)
|
||||||
|
|
||||||
|
def first_event(self, earliest: dt.datetime = None) -> Optional[dt.datetime]:
|
||||||
|
"""Date/Time of first event. Returns `None` if series has no events.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
- earliest: Date of first events to count(inclusive), or any if not specified
|
||||||
|
"""
|
||||||
|
min = "-inf" if not earliest else earliest.timestamp()
|
||||||
|
event = self._redis.zrangebyscore(
|
||||||
|
self._key_sorted_set,
|
||||||
|
min,
|
||||||
|
"+inf",
|
||||||
|
withscores=True,
|
||||||
|
start=0,
|
||||||
|
num=1,
|
||||||
|
score_cast_func=self._cast_scores_to_dt,
|
||||||
|
)
|
||||||
|
if not event:
|
||||||
|
return None
|
||||||
|
return event[0][1]
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def _cast_scores_to_dt(score) -> dt.datetime:
|
||||||
|
return dt.datetime.fromtimestamp(float(score), tz=utc)
|
||||||
54
allianceauth/authentication/task_statistics/signals.py
Normal file
54
allianceauth/authentication/task_statistics/signals.py
Normal file
@@ -0,0 +1,54 @@
|
|||||||
|
from celery.signals import (
|
||||||
|
task_failure,
|
||||||
|
task_internal_error,
|
||||||
|
task_retry,
|
||||||
|
task_success,
|
||||||
|
worker_ready
|
||||||
|
)
|
||||||
|
|
||||||
|
from django.conf import settings
|
||||||
|
|
||||||
|
from .counters import failed_tasks, retried_tasks, succeeded_tasks
|
||||||
|
|
||||||
|
|
||||||
|
def reset_counters():
|
||||||
|
"""Reset all counters for the celery status."""
|
||||||
|
succeeded_tasks.clear()
|
||||||
|
failed_tasks.clear()
|
||||||
|
retried_tasks.clear()
|
||||||
|
|
||||||
|
|
||||||
|
def is_enabled() -> bool:
|
||||||
|
return not bool(
|
||||||
|
getattr(settings, "ALLIANCEAUTH_DASHBOARD_TASK_STATISTICS_DISABLED", False)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@worker_ready.connect
|
||||||
|
def reset_counters_when_celery_restarted(*args, **kwargs):
|
||||||
|
if is_enabled():
|
||||||
|
reset_counters()
|
||||||
|
|
||||||
|
|
||||||
|
@task_success.connect
|
||||||
|
def record_task_succeeded(*args, **kwargs):
|
||||||
|
if is_enabled():
|
||||||
|
succeeded_tasks.add()
|
||||||
|
|
||||||
|
|
||||||
|
@task_retry.connect
|
||||||
|
def record_task_retried(*args, **kwargs):
|
||||||
|
if is_enabled():
|
||||||
|
retried_tasks.add()
|
||||||
|
|
||||||
|
|
||||||
|
@task_failure.connect
|
||||||
|
def record_task_failed(*args, **kwargs):
|
||||||
|
if is_enabled():
|
||||||
|
failed_tasks.add()
|
||||||
|
|
||||||
|
|
||||||
|
@task_internal_error.connect
|
||||||
|
def record_task_internal_error(*args, **kwargs):
|
||||||
|
if is_enabled():
|
||||||
|
failed_tasks.add()
|
||||||
@@ -0,0 +1,51 @@
|
|||||||
|
import datetime as dt
|
||||||
|
|
||||||
|
from django.test import TestCase
|
||||||
|
from django.utils.timezone import now
|
||||||
|
|
||||||
|
from allianceauth.authentication.task_statistics.counters import (
|
||||||
|
dashboard_results,
|
||||||
|
succeeded_tasks,
|
||||||
|
retried_tasks,
|
||||||
|
failed_tasks,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class TestDashboardResults(TestCase):
|
||||||
|
def test_should_return_counts_for_given_timeframe_only(self):
|
||||||
|
# given
|
||||||
|
earliest_task = now() - dt.timedelta(minutes=15)
|
||||||
|
succeeded_tasks.clear()
|
||||||
|
succeeded_tasks.add(now() - dt.timedelta(hours=1, seconds=1))
|
||||||
|
succeeded_tasks.add(earliest_task)
|
||||||
|
succeeded_tasks.add()
|
||||||
|
succeeded_tasks.add()
|
||||||
|
retried_tasks.clear()
|
||||||
|
retried_tasks.add(now() - dt.timedelta(hours=1, seconds=1))
|
||||||
|
retried_tasks.add(now() - dt.timedelta(seconds=30))
|
||||||
|
retried_tasks.add()
|
||||||
|
failed_tasks.clear()
|
||||||
|
failed_tasks.add(now() - dt.timedelta(hours=1, seconds=1))
|
||||||
|
failed_tasks.add()
|
||||||
|
# when
|
||||||
|
results = dashboard_results(hours=1)
|
||||||
|
# then
|
||||||
|
self.assertEqual(results.succeeded, 3)
|
||||||
|
self.assertEqual(results.retried, 2)
|
||||||
|
self.assertEqual(results.failed, 1)
|
||||||
|
self.assertEqual(results.total, 6)
|
||||||
|
self.assertEqual(results.earliest_task, earliest_task)
|
||||||
|
|
||||||
|
def test_should_work_with_no_data(self):
|
||||||
|
# given
|
||||||
|
succeeded_tasks.clear()
|
||||||
|
retried_tasks.clear()
|
||||||
|
failed_tasks.clear()
|
||||||
|
# when
|
||||||
|
results = dashboard_results(hours=1)
|
||||||
|
# then
|
||||||
|
self.assertEqual(results.succeeded, 0)
|
||||||
|
self.assertEqual(results.retried, 0)
|
||||||
|
self.assertEqual(results.failed, 0)
|
||||||
|
self.assertEqual(results.total, 0)
|
||||||
|
self.assertIsNone(results.earliest_task)
|
||||||
@@ -0,0 +1,133 @@
|
|||||||
|
import datetime as dt
|
||||||
|
|
||||||
|
from pytz import utc
|
||||||
|
from django.test import TestCase
|
||||||
|
from django.utils.timezone import now
|
||||||
|
|
||||||
|
from allianceauth.authentication.task_statistics.event_series import EventSeries
|
||||||
|
|
||||||
|
|
||||||
|
class TestEventSeries(TestCase):
|
||||||
|
def test_should_add_event(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
# when
|
||||||
|
events.add()
|
||||||
|
# then
|
||||||
|
result = events.all()
|
||||||
|
self.assertEqual(len(result), 1)
|
||||||
|
self.assertAlmostEqual(result[0], now(), delta=dt.timedelta(seconds=30))
|
||||||
|
|
||||||
|
def test_should_add_event_with_specified_time(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
my_time = dt.datetime(2021, 11, 1, 12, 15, tzinfo=utc)
|
||||||
|
# when
|
||||||
|
events.add(my_time)
|
||||||
|
# then
|
||||||
|
result = events.all()
|
||||||
|
self.assertEqual(len(result), 1)
|
||||||
|
self.assertAlmostEqual(result[0], my_time, delta=dt.timedelta(seconds=30))
|
||||||
|
|
||||||
|
def test_should_count_events(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
events.add()
|
||||||
|
events.add()
|
||||||
|
# when
|
||||||
|
result = events.count()
|
||||||
|
# then
|
||||||
|
self.assertEqual(result, 2)
|
||||||
|
|
||||||
|
def test_should_count_zero(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
# when
|
||||||
|
result = events.count()
|
||||||
|
# then
|
||||||
|
self.assertEqual(result, 0)
|
||||||
|
|
||||||
|
def test_should_count_events_within_timeframe_1(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 0, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 10, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 15, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 30, tzinfo=utc))
|
||||||
|
# when
|
||||||
|
result = events.count(
|
||||||
|
earliest=dt.datetime(2021, 12, 1, 12, 8, tzinfo=utc),
|
||||||
|
latest=dt.datetime(2021, 12, 1, 12, 17, tzinfo=utc),
|
||||||
|
)
|
||||||
|
# then
|
||||||
|
self.assertEqual(result, 2)
|
||||||
|
|
||||||
|
def test_should_count_events_within_timeframe_2(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 0, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 10, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 15, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 30, tzinfo=utc))
|
||||||
|
# when
|
||||||
|
result = events.count(earliest=dt.datetime(2021, 12, 1, 12, 8))
|
||||||
|
# then
|
||||||
|
self.assertEqual(result, 3)
|
||||||
|
|
||||||
|
def test_should_count_events_within_timeframe_3(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 0, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 10, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 15, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 30, tzinfo=utc))
|
||||||
|
# when
|
||||||
|
result = events.count(latest=dt.datetime(2021, 12, 1, 12, 12))
|
||||||
|
# then
|
||||||
|
self.assertEqual(result, 2)
|
||||||
|
|
||||||
|
def test_should_clear_events(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
events.add()
|
||||||
|
events.add()
|
||||||
|
# when
|
||||||
|
events.clear()
|
||||||
|
# then
|
||||||
|
self.assertEqual(events.count(), 0)
|
||||||
|
|
||||||
|
def test_should_return_date_of_first_event(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 0, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 10, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 15, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 30, tzinfo=utc))
|
||||||
|
# when
|
||||||
|
result = events.first_event()
|
||||||
|
# then
|
||||||
|
self.assertEqual(result, dt.datetime(2021, 12, 1, 12, 0, tzinfo=utc))
|
||||||
|
|
||||||
|
def test_should_return_date_of_first_event_with_range(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 0, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 10, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 15, tzinfo=utc))
|
||||||
|
events.add(dt.datetime(2021, 12, 1, 12, 30, tzinfo=utc))
|
||||||
|
# when
|
||||||
|
result = events.first_event(
|
||||||
|
earliest=dt.datetime(2021, 12, 1, 12, 8, tzinfo=utc)
|
||||||
|
)
|
||||||
|
# then
|
||||||
|
self.assertEqual(result, dt.datetime(2021, 12, 1, 12, 10, tzinfo=utc))
|
||||||
|
|
||||||
|
def test_should_return_all_events(self):
|
||||||
|
# given
|
||||||
|
events = EventSeries("dummy")
|
||||||
|
events.add()
|
||||||
|
events.add()
|
||||||
|
# when
|
||||||
|
results = events.all()
|
||||||
|
# then
|
||||||
|
self.assertEqual(len(results), 2)
|
||||||
@@ -0,0 +1,93 @@
|
|||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
from celery.exceptions import Retry
|
||||||
|
|
||||||
|
from django.test import TestCase, override_settings
|
||||||
|
|
||||||
|
from allianceauth.authentication.task_statistics.counters import (
|
||||||
|
failed_tasks,
|
||||||
|
retried_tasks,
|
||||||
|
succeeded_tasks,
|
||||||
|
)
|
||||||
|
from allianceauth.authentication.task_statistics.signals import (
|
||||||
|
reset_counters,
|
||||||
|
is_enabled,
|
||||||
|
)
|
||||||
|
from allianceauth.eveonline.tasks import update_character
|
||||||
|
|
||||||
|
|
||||||
|
@override_settings(
|
||||||
|
CELERY_ALWAYS_EAGER=True,ALLIANCEAUTH_DASHBOARD_TASK_STATISTICS_DISABLED=False
|
||||||
|
)
|
||||||
|
class TestTaskSignals(TestCase):
|
||||||
|
fixtures = ["disable_analytics"]
|
||||||
|
|
||||||
|
def test_should_record_successful_task(self):
|
||||||
|
# given
|
||||||
|
succeeded_tasks.clear()
|
||||||
|
retried_tasks.clear()
|
||||||
|
failed_tasks.clear()
|
||||||
|
# when
|
||||||
|
with patch(
|
||||||
|
"allianceauth.eveonline.tasks.EveCharacter.objects.update_character"
|
||||||
|
) as mock_update:
|
||||||
|
mock_update.return_value = None
|
||||||
|
update_character.delay(1)
|
||||||
|
# then
|
||||||
|
self.assertEqual(succeeded_tasks.count(), 1)
|
||||||
|
self.assertEqual(retried_tasks.count(), 0)
|
||||||
|
self.assertEqual(failed_tasks.count(), 0)
|
||||||
|
|
||||||
|
def test_should_record_retried_task(self):
|
||||||
|
# given
|
||||||
|
succeeded_tasks.clear()
|
||||||
|
retried_tasks.clear()
|
||||||
|
failed_tasks.clear()
|
||||||
|
# when
|
||||||
|
with patch(
|
||||||
|
"allianceauth.eveonline.tasks.EveCharacter.objects.update_character"
|
||||||
|
) as mock_update:
|
||||||
|
mock_update.side_effect = Retry
|
||||||
|
update_character.delay(1)
|
||||||
|
# then
|
||||||
|
self.assertEqual(succeeded_tasks.count(), 0)
|
||||||
|
self.assertEqual(failed_tasks.count(), 0)
|
||||||
|
self.assertEqual(retried_tasks.count(), 1)
|
||||||
|
|
||||||
|
def test_should_record_failed_task(self):
|
||||||
|
# given
|
||||||
|
succeeded_tasks.clear()
|
||||||
|
retried_tasks.clear()
|
||||||
|
failed_tasks.clear()
|
||||||
|
# when
|
||||||
|
with patch(
|
||||||
|
"allianceauth.eveonline.tasks.EveCharacter.objects.update_character"
|
||||||
|
) as mock_update:
|
||||||
|
mock_update.side_effect = RuntimeError
|
||||||
|
update_character.delay(1)
|
||||||
|
# then
|
||||||
|
self.assertEqual(succeeded_tasks.count(), 0)
|
||||||
|
self.assertEqual(retried_tasks.count(), 0)
|
||||||
|
self.assertEqual(failed_tasks.count(), 1)
|
||||||
|
|
||||||
|
def test_should_reset_counters(self):
|
||||||
|
# given
|
||||||
|
succeeded_tasks.add()
|
||||||
|
retried_tasks.add()
|
||||||
|
failed_tasks.add()
|
||||||
|
# when
|
||||||
|
reset_counters()
|
||||||
|
# then
|
||||||
|
self.assertEqual(succeeded_tasks.count(), 0)
|
||||||
|
self.assertEqual(retried_tasks.count(), 0)
|
||||||
|
self.assertEqual(failed_tasks.count(), 0)
|
||||||
|
|
||||||
|
|
||||||
|
class TestIsEnabled(TestCase):
|
||||||
|
@override_settings(ALLIANCEAUTH_DASHBOARD_TASK_STATISTICS_DISABLED=False)
|
||||||
|
def test_enabled(self):
|
||||||
|
self.assertTrue(is_enabled())
|
||||||
|
|
||||||
|
@override_settings(ALLIANCEAUTH_DASHBOARD_TASK_STATISTICS_DISABLED=True)
|
||||||
|
def test_disabled(self):
|
||||||
|
self.assertFalse(is_enabled())
|
||||||
@@ -22,13 +22,13 @@ def check_character_ownership(owner_hash):
|
|||||||
continue
|
continue
|
||||||
except (KeyError, IncompleteResponseError):
|
except (KeyError, IncompleteResponseError):
|
||||||
# We can't validate the hash hasn't changed but also can't assume it has. Abort for now.
|
# We can't validate the hash hasn't changed but also can't assume it has. Abort for now.
|
||||||
logger.warning("Failed to validate owner hash of {0} due to problems contacting SSO servers.".format(
|
logger.warning("Failed to validate owner hash of {} due to problems contacting SSO servers.".format(
|
||||||
tokens[0].character_name))
|
tokens[0].character_name))
|
||||||
break
|
break
|
||||||
|
|
||||||
if not t.character_owner_hash == old_hash:
|
if not t.character_owner_hash == old_hash:
|
||||||
logger.info(
|
logger.info(
|
||||||
'Character %s has changed ownership. Revoking %s tokens.' % (t.character_name, tokens.count()))
|
f'Character {t.character_name} has changed ownership. Revoking {tokens.count()} tokens.')
|
||||||
tokens.delete()
|
tokens.delete()
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|||||||
@@ -2,10 +2,10 @@
|
|||||||
{% load static %}
|
{% load static %}
|
||||||
{% load i18n %}
|
{% load i18n %}
|
||||||
|
|
||||||
{% block page_title %}{% trans "Dashboard" %}{% endblock %}
|
{% block page_title %}{% translate "Dashboard" %}{% endblock %}
|
||||||
|
|
||||||
{% block content %}
|
{% block content %}
|
||||||
<h1 class="page-header text-center">{% trans "Dashboard" %}</h1>
|
<h1 class="page-header text-center">{% translate "Dashboard" %}</h1>
|
||||||
{% if user.is_staff %}
|
{% if user.is_staff %}
|
||||||
{% include 'allianceauth/admin-status/include.html' %}
|
{% include 'allianceauth/admin-status/include.html' %}
|
||||||
{% endif %}
|
{% endif %}
|
||||||
@@ -60,6 +60,17 @@
|
|||||||
<td class="text-center">{{ main.alliance_name }}</td>
|
<td class="text-center">{{ main.alliance_name }}</td>
|
||||||
<tr>
|
<tr>
|
||||||
</table>
|
</table>
|
||||||
|
{% elif main.faction_id %}
|
||||||
|
<table class="table">
|
||||||
|
<tr>
|
||||||
|
<td class="text-center">
|
||||||
|
<img class="ra-avatar"src="{{ main.faction_logo_url_128 }}">
|
||||||
|
</td>
|
||||||
|
</tr>
|
||||||
|
<tr>
|
||||||
|
<td class="text-center">{{ main.faction_name }}</td>
|
||||||
|
<tr>
|
||||||
|
</table>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -67,29 +78,39 @@
|
|||||||
<p>
|
<p>
|
||||||
<img class="ra-avatar" src="{{ main.portrait_url_64 }}">
|
<img class="ra-avatar" src="{{ main.portrait_url_64 }}">
|
||||||
<img class="ra-avatar" src="{{ main.corporation_logo_url_64 }}">
|
<img class="ra-avatar" src="{{ main.corporation_logo_url_64 }}">
|
||||||
|
{% if main.alliance_id %}
|
||||||
<img class="ra-avatar" src="{{ main.alliance_logo_url_64 }}">
|
<img class="ra-avatar" src="{{ main.alliance_logo_url_64 }}">
|
||||||
|
{% endif %}
|
||||||
|
{% if main.faction_id %}
|
||||||
|
<img class="ra-avatar" src="{{ main.faction_logo_url_64 }}">
|
||||||
|
{% endif %}
|
||||||
</p>
|
</p>
|
||||||
<p>
|
<p>
|
||||||
<strong>{{ main.character_name }}</strong><br>
|
<strong>{{ main.character_name }}</strong><br>
|
||||||
{{ main.corporation_name }}<br>
|
{{ main.corporation_name }}<br>
|
||||||
{{ main.alliance_name }}
|
{% if main.alliance_id %}
|
||||||
|
{{ main.alliance_name }}<br>
|
||||||
|
{% endif %}
|
||||||
|
{% if main.faction_id %}
|
||||||
|
{{ main.faction_name }}
|
||||||
|
{% endif %}
|
||||||
</p>
|
</p>
|
||||||
</div>
|
</div>
|
||||||
{% endwith %}
|
{% endwith %}
|
||||||
{% else %}
|
{% else %}
|
||||||
<div class="alert alert-danger" role="alert">
|
<div class="alert alert-danger" role="alert">
|
||||||
{% trans "No main character set." %}
|
{% translate "No main character set." %}
|
||||||
</div>
|
</div>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
<div class="clearfix"></div>
|
<div class="clearfix"></div>
|
||||||
<div class="row">
|
<div class="row">
|
||||||
<div class="col-sm-6 button-wrapper">
|
<div class="col-sm-6 button-wrapper">
|
||||||
<a href="{% url 'authentication:add_character' %}" class="btn btn-block btn-info"
|
<a href="{% url 'authentication:add_character' %}" class="btn btn-block btn-info"
|
||||||
title="Add Character">{% trans 'Add Character' %}</a>
|
title="Add Character">{% translate 'Add Character' %}</a>
|
||||||
</div>
|
</div>
|
||||||
<div class="col-sm-6 button-wrapper">
|
<div class="col-sm-6 button-wrapper">
|
||||||
<a href="{% url 'authentication:change_main_character' %}" class="btn btn-block btn-info"
|
<a href="{% url 'authentication:change_main_character' %}" class="btn btn-block btn-info"
|
||||||
title="Change Main Character">{% trans "Change Main" %}</a>
|
title="Change Main Character">{% translate "Change Main" %}</a>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -98,7 +119,7 @@
|
|||||||
<div class="col-sm-6 text-center">
|
<div class="col-sm-6 text-center">
|
||||||
<div class="panel panel-success" style="height:100%">
|
<div class="panel panel-success" style="height:100%">
|
||||||
<div class="panel-heading">
|
<div class="panel-heading">
|
||||||
<h3 class="panel-title">{% trans "Group Memberships" %}</h3>
|
<h3 class="panel-title">{% translate "Group Memberships" %}</h3>
|
||||||
</div>
|
</div>
|
||||||
<div class="panel-body">
|
<div class="panel-body">
|
||||||
<div style="height: 240px;overflow:-moz-scrollbars-vertical;overflow-y:auto;">
|
<div style="height: 240px;overflow:-moz-scrollbars-vertical;overflow-y:auto;">
|
||||||
@@ -118,7 +139,7 @@
|
|||||||
<div class="panel panel-default">
|
<div class="panel panel-default">
|
||||||
<div class="panel-heading">
|
<div class="panel-heading">
|
||||||
<h3 class="panel-title text-center" style="text-align: center">
|
<h3 class="panel-title text-center" style="text-align: center">
|
||||||
{% trans 'Characters' %}
|
{% translate 'Characters' %}
|
||||||
</h3>
|
</h3>
|
||||||
</div>
|
</div>
|
||||||
<div class="panel-body">
|
<div class="panel-body">
|
||||||
@@ -126,9 +147,9 @@
|
|||||||
<thead>
|
<thead>
|
||||||
<tr>
|
<tr>
|
||||||
<th class="text-center"></th>
|
<th class="text-center"></th>
|
||||||
<th class="text-center">{% trans 'Name' %}</th>
|
<th class="text-center">{% translate 'Name' %}</th>
|
||||||
<th class="text-center">{% trans 'Corp' %}</th>
|
<th class="text-center">{% translate 'Corp' %}</th>
|
||||||
<th class="text-center">{% trans 'Alliance' %}</th>
|
<th class="text-center">{% translate 'Alliance' %}</th>
|
||||||
</tr>
|
</tr>
|
||||||
</thead>
|
</thead>
|
||||||
<tbody>
|
<tbody>
|
||||||
|
|||||||
@@ -12,7 +12,7 @@
|
|||||||
|
|
||||||
{% include 'allianceauth/icons.html' %}
|
{% include 'allianceauth/icons.html' %}
|
||||||
|
|
||||||
<title>{% block title %}{{ SITE_NAME }}{% endblock %}</title>
|
<title>{% block title %}{% block page_title %}{% endblock page_title %} - {{ SITE_NAME }}{% endblock title %}</title>
|
||||||
|
|
||||||
{% include 'bundles/bootstrap-css.html' %}
|
{% include 'bundles/bootstrap-css.html' %}
|
||||||
{% include 'bundles/fontawesome.html' %}
|
{% include 'bundles/fontawesome.html' %}
|
||||||
|
|||||||
@@ -1,6 +1,10 @@
|
|||||||
{% extends 'public/middle_box.html' %}
|
{% extends 'public/middle_box.html' %}
|
||||||
|
|
||||||
|
{% load i18n %}
|
||||||
{% load static %}
|
{% load static %}
|
||||||
{% block page_title %}Login{% endblock %}
|
|
||||||
|
{% block page_title %}{% translate "Login" %}{% endblock %}
|
||||||
|
|
||||||
{% block middle_box_content %}
|
{% block middle_box_content %}
|
||||||
<a href="{% url 'auth_sso_login' %}{% if request.GET.next %}?next={{request.GET.next}}{%endif%}">
|
<a href="{% url 'auth_sso_login' %}{% if request.GET.next %}?next={{request.GET.next}}{%endif%}">
|
||||||
<img class="img-responsive center-block" src="{% static 'img/sso/EVE_SSO_Login_Buttons_Large_Black.png' %}" border=0>
|
<img class="img-responsive center-block" src="{% static 'img/sso/EVE_SSO_Login_Buttons_Large_Black.png' %}" border=0>
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
{% extends 'public/base.html' %}
|
{% extends 'public/base.html' %}
|
||||||
{% load static %}
|
{% load static %}
|
||||||
{% block title %}Login{% endblock %}
|
|
||||||
{% block content %}
|
{% block content %}
|
||||||
<div class="col-md-4 col-md-offset-4">
|
<div class="col-md-4 col-md-offset-4">
|
||||||
{% if messages %}
|
{% if messages %}
|
||||||
|
|||||||
@@ -1,13 +1,17 @@
|
|||||||
{% extends 'public/base.html' %}
|
{% extends 'public/base.html' %}
|
||||||
|
|
||||||
{% load static %}
|
{% load static %}
|
||||||
{% load bootstrap %}
|
{% load bootstrap %}
|
||||||
{% load i18n %}
|
{% load i18n %}
|
||||||
{% block page_title %}Registration{% endblock %}
|
|
||||||
|
{% block page_title %}{% translate "Registration" %}{% endblock %}
|
||||||
|
|
||||||
{% block extra_include %}
|
{% block extra_include %}
|
||||||
{% include 'bundles/bootstrap-css.html' %}
|
{% include 'bundles/bootstrap-css.html' %}
|
||||||
{% include 'bundles/fontawesome.html' %}
|
{% include 'bundles/fontawesome.html' %}
|
||||||
{% include 'bundles/bootstrap-js.html' %}
|
{% include 'bundles/bootstrap-js.html' %}
|
||||||
{% endblock %}
|
{% endblock %}
|
||||||
|
|
||||||
{% block content %}
|
{% block content %}
|
||||||
<div class="col-md-4 col-md-offset-4">
|
<div class="col-md-4 col-md-offset-4">
|
||||||
<div class="panel panel-default panel-transparent">
|
<div class="panel panel-default panel-transparent">
|
||||||
@@ -15,7 +19,7 @@
|
|||||||
<form method="POST">
|
<form method="POST">
|
||||||
{% csrf_token %}
|
{% csrf_token %}
|
||||||
{{ form|bootstrap }}
|
{{ form|bootstrap }}
|
||||||
<button class="btn btn-lg btn-primary btn-block" type="submit">{% trans "Register" %}</button>
|
<button class="btn btn-lg btn-primary btn-block" type="submit">{% translate "Register" %}</button>
|
||||||
</form>
|
</form>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
{% extends 'public/middle_box.html' %}
|
{% extends 'public/middle_box.html' %}
|
||||||
{% load i18n %}
|
{% load i18n %}
|
||||||
{% block middle_box_content %}
|
{% block middle_box_content %}
|
||||||
<div class="alert alert-danger">{% trans 'Invalid or expired activation link.' %}</div>
|
<div class="alert alert-danger">{% translate 'Invalid or expired activation link.' %}</div>
|
||||||
{% endblock %}
|
{% endblock %}
|
||||||
|
|||||||
@@ -2,10 +2,6 @@ You're receiving this email because someone has entered this email address while
|
|||||||
|
|
||||||
If this was you, please click on the link below to confirm your email address:
|
If this was you, please click on the link below to confirm your email address:
|
||||||
|
|
||||||
<a href="{{ scheme }}://{{ url }}">Confirm email address</a>
|
|
||||||
|
|
||||||
Link not working? Try copy/pasting this URL into your browser:
|
|
||||||
|
|
||||||
{{ scheme }}://{{ url }}
|
{{ scheme }}://{{ url }}
|
||||||
|
|
||||||
This link will expire in {{ expiration_days }} day(s).
|
This link will expire in {{ expiration_days }} day(s).
|
||||||
|
|||||||
@@ -0,0 +1,19 @@
|
|||||||
|
<p>
|
||||||
|
You're receiving this email because someone has entered this email address while registering for an account on {{ site.domain }}
|
||||||
|
</p>
|
||||||
|
|
||||||
|
<p>
|
||||||
|
If this was you, please click on the link below to confirm your email address:
|
||||||
|
<p>
|
||||||
|
|
||||||
|
<p>
|
||||||
|
<a href="{{ scheme }}://{{ url }}">Confirm email address</a>
|
||||||
|
</p>
|
||||||
|
|
||||||
|
<p>
|
||||||
|
This link will expire in {{ expiration_days }} day(s).
|
||||||
|
</p>
|
||||||
|
|
||||||
|
<p>
|
||||||
|
If this was not you, it is safe to ignore this email.
|
||||||
|
</p>
|
||||||
@@ -2,13 +2,13 @@
|
|||||||
{% blocktrans trimmed %}You're receiving this email because you requested a password reset for your
|
{% blocktrans trimmed %}You're receiving this email because you requested a password reset for your
|
||||||
user account.{% endblocktrans %}
|
user account.{% endblocktrans %}
|
||||||
|
|
||||||
{% trans "Please go to the following page and choose a new password:" %}
|
{% translate "Please go to the following page and choose a new password:" %}
|
||||||
{% block reset_link %}
|
{% block reset_link %}
|
||||||
{{domain}}{% url 'password_reset_confirm' uidb64=uid token=token %}
|
{{domain}}{% url 'password_reset_confirm' uidb64=uid token=token %}
|
||||||
{% endblock %}
|
{% endblock %}
|
||||||
{% trans "Your username, in case you've forgotten:" %} {{ user.get_username }}
|
{% translate "Your username, in case you've forgotten:" %} {{ user.get_username }}
|
||||||
|
|
||||||
{% trans "Thanks for using our site!" %}
|
{% translate "Thanks for using our site!" %}
|
||||||
|
|
||||||
{% blocktrans %}Your IT Team{% endblocktrans %}
|
{% blocktrans %}Your IT Team{% endblocktrans %}
|
||||||
|
|
||||||
|
|||||||
@@ -2,13 +2,13 @@
|
|||||||
{% load bootstrap %}
|
{% load bootstrap %}
|
||||||
{% load i18n %}
|
{% load i18n %}
|
||||||
{% load static %}
|
{% load static %}
|
||||||
{% block page_title %}Register{% endblock %}
|
{% block page_title %}{% translate "Register" %}{% endblock %}
|
||||||
{% block middle_box_content %}
|
{% block middle_box_content %}
|
||||||
<form class="form-signin" role="form" action="" method="POST">
|
<form class="form-signin" role="form" action="" method="POST">
|
||||||
{% csrf_token %}
|
{% csrf_token %}
|
||||||
{{ form|bootstrap }}
|
{{ form|bootstrap }}
|
||||||
<br/>
|
<br/>
|
||||||
<button class="btn btn-lg btn-primary btn-block" type="submit">{% trans "Submit" %}</button>
|
<button class="btn btn-lg btn-primary btn-block" type="submit">{% translate "Submit" %}</button>
|
||||||
<br/>
|
<br/>
|
||||||
</form>
|
</form>
|
||||||
{% endblock %}
|
{% endblock %}
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
from bs4 import BeautifulSoup
|
||||||
from urllib.parse import quote
|
from urllib.parse import quote
|
||||||
from unittest.mock import patch, MagicMock
|
from unittest.mock import patch, MagicMock
|
||||||
|
|
||||||
@@ -9,7 +10,7 @@ from allianceauth.authentication.models import (
|
|||||||
CharacterOwnership, State, OwnershipRecord
|
CharacterOwnership, State, OwnershipRecord
|
||||||
)
|
)
|
||||||
from allianceauth.eveonline.models import (
|
from allianceauth.eveonline.models import (
|
||||||
EveCharacter, EveCorporationInfo, EveAllianceInfo
|
EveCharacter, EveCorporationInfo, EveAllianceInfo, EveFactionInfo
|
||||||
)
|
)
|
||||||
from allianceauth.services.hooks import ServicesHook
|
from allianceauth.services.hooks import ServicesHook
|
||||||
from allianceauth.tests.auth_utils import AuthUtils
|
from allianceauth.tests.auth_utils import AuthUtils
|
||||||
@@ -20,6 +21,7 @@ from ..admin import (
|
|||||||
StateAdmin,
|
StateAdmin,
|
||||||
MainCorporationsFilter,
|
MainCorporationsFilter,
|
||||||
MainAllianceFilter,
|
MainAllianceFilter,
|
||||||
|
MainFactionFilter,
|
||||||
OwnershipRecordAdmin,
|
OwnershipRecordAdmin,
|
||||||
User,
|
User,
|
||||||
UserAdmin,
|
UserAdmin,
|
||||||
@@ -36,7 +38,7 @@ from . import get_admin_change_view_url, get_admin_search_url
|
|||||||
MODULE_PATH = 'allianceauth.authentication.admin'
|
MODULE_PATH = 'allianceauth.authentication.admin'
|
||||||
|
|
||||||
|
|
||||||
class MockRequest(object):
|
class MockRequest:
|
||||||
def __init__(self, user=None):
|
def __init__(self, user=None):
|
||||||
self.user = user
|
self.user = user
|
||||||
|
|
||||||
@@ -180,6 +182,31 @@ class TestCaseWithTestData(TestCase):
|
|||||||
cls.user_3.is_superuser = True
|
cls.user_3.is_superuser = True
|
||||||
cls.user_3.save()
|
cls.user_3.save()
|
||||||
|
|
||||||
|
# user 4 - corp and faction, no alliance
|
||||||
|
cls.character_4 = EveCharacter.objects.create(
|
||||||
|
character_id=4321,
|
||||||
|
character_name='Professor X',
|
||||||
|
corporation_id=5432,
|
||||||
|
corporation_name="Xavier's School for Gifted Youngsters",
|
||||||
|
corporation_ticker='MUTNT',
|
||||||
|
alliance_id=None,
|
||||||
|
faction_id=999,
|
||||||
|
faction_name='The X-Men',
|
||||||
|
)
|
||||||
|
cls.user_4 = User.objects.create_user(
|
||||||
|
cls.character_4.character_name.replace(' ', '_'),
|
||||||
|
'abc@example.com',
|
||||||
|
'password'
|
||||||
|
)
|
||||||
|
CharacterOwnership.objects.create(
|
||||||
|
character=cls.character_4,
|
||||||
|
owner_hash='x1' + cls.character_4.character_name,
|
||||||
|
user=cls.user_4
|
||||||
|
)
|
||||||
|
cls.user_4.profile.main_character = cls.character_4
|
||||||
|
cls.user_4.profile.save()
|
||||||
|
EveFactionInfo.objects.create(faction_id=999, faction_name='The X-Men')
|
||||||
|
|
||||||
|
|
||||||
def make_generic_search_request(ModelClass: type, search_term: str):
|
def make_generic_search_request(ModelClass: type, search_term: str):
|
||||||
User.objects.create_superuser(
|
User.objects.create_superuser(
|
||||||
@@ -188,11 +215,12 @@ def make_generic_search_request(ModelClass: type, search_term: str):
|
|||||||
c = Client()
|
c = Client()
|
||||||
c.login(username='superuser', password='secret')
|
c.login(username='superuser', password='secret')
|
||||||
return c.get(
|
return c.get(
|
||||||
'%s?q=%s' % (get_admin_search_url(ModelClass), quote(search_term))
|
f'{get_admin_search_url(ModelClass)}?q={quote(search_term)}'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
class TestCharacterOwnershipAdmin(TestCaseWithTestData):
|
class TestCharacterOwnershipAdmin(TestCaseWithTestData):
|
||||||
|
fixtures = ["disable_analytics"]
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.modeladmin = CharacterOwnershipAdmin(
|
self.modeladmin = CharacterOwnershipAdmin(
|
||||||
@@ -219,6 +247,7 @@ class TestCharacterOwnershipAdmin(TestCaseWithTestData):
|
|||||||
|
|
||||||
|
|
||||||
class TestOwnershipRecordAdmin(TestCaseWithTestData):
|
class TestOwnershipRecordAdmin(TestCaseWithTestData):
|
||||||
|
fixtures = ["disable_analytics"]
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.modeladmin = OwnershipRecordAdmin(
|
self.modeladmin = OwnershipRecordAdmin(
|
||||||
@@ -245,6 +274,7 @@ class TestOwnershipRecordAdmin(TestCaseWithTestData):
|
|||||||
|
|
||||||
|
|
||||||
class TestStateAdmin(TestCaseWithTestData):
|
class TestStateAdmin(TestCaseWithTestData):
|
||||||
|
fixtures = ["disable_analytics"]
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.modeladmin = StateAdmin(
|
self.modeladmin = StateAdmin(
|
||||||
@@ -274,6 +304,7 @@ class TestStateAdmin(TestCaseWithTestData):
|
|||||||
|
|
||||||
|
|
||||||
class TestUserAdmin(TestCaseWithTestData):
|
class TestUserAdmin(TestCaseWithTestData):
|
||||||
|
fixtures = ["disable_analytics"]
|
||||||
|
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.factory = RequestFactory()
|
self.factory = RequestFactory()
|
||||||
@@ -315,9 +346,13 @@ class TestUserAdmin(TestCaseWithTestData):
|
|||||||
self.assertEqual(user_main_organization(self.user_2), expected)
|
self.assertEqual(user_main_organization(self.user_2), expected)
|
||||||
|
|
||||||
def test_user_main_organization_u3(self):
|
def test_user_main_organization_u3(self):
|
||||||
expected = None
|
expected = ''
|
||||||
self.assertEqual(user_main_organization(self.user_3), expected)
|
self.assertEqual(user_main_organization(self.user_3), expected)
|
||||||
|
|
||||||
|
def test_user_main_organization_u4(self):
|
||||||
|
expected = "Xavier's School for Gifted Youngsters<br>The X-Men"
|
||||||
|
self.assertEqual(user_main_organization(self.user_4), expected)
|
||||||
|
|
||||||
def test_characters_u1(self):
|
def test_characters_u1(self):
|
||||||
expected = 'Batman, Bruce Wayne'
|
expected = 'Batman, Bruce Wayne'
|
||||||
result = self.modeladmin._characters(self.user_1)
|
result = self.modeladmin._characters(self.user_1)
|
||||||
@@ -390,7 +425,7 @@ class TestUserAdmin(TestCaseWithTestData):
|
|||||||
|
|
||||||
# actions
|
# actions
|
||||||
|
|
||||||
@patch(MODULE_PATH + '.UserAdmin.message_user', auto_spec=True)
|
@patch(MODULE_PATH + '.UserAdmin.message_user', auto_spec=True, unsafe=True)
|
||||||
@patch(MODULE_PATH + '.update_character')
|
@patch(MODULE_PATH + '.update_character')
|
||||||
def test_action_update_main_character_model(
|
def test_action_update_main_character_model(
|
||||||
self, mock_task, mock_message_user
|
self, mock_task, mock_message_user
|
||||||
@@ -420,6 +455,7 @@ class TestUserAdmin(TestCaseWithTestData):
|
|||||||
expected = [
|
expected = [
|
||||||
(2002, 'Daily Planet'),
|
(2002, 'Daily Planet'),
|
||||||
(2001, 'Wayne Technologies'),
|
(2001, 'Wayne Technologies'),
|
||||||
|
(5432, "Xavier's School for Gifted Youngsters"),
|
||||||
]
|
]
|
||||||
self.assertEqual(filterspec.lookup_choices, expected)
|
self.assertEqual(filterspec.lookup_choices, expected)
|
||||||
|
|
||||||
@@ -463,6 +499,34 @@ class TestUserAdmin(TestCaseWithTestData):
|
|||||||
expected = [self.user_1]
|
expected = [self.user_1]
|
||||||
self.assertSetEqual(set(queryset), set(expected))
|
self.assertSetEqual(set(queryset), set(expected))
|
||||||
|
|
||||||
|
def test_filter_main_factions(self):
|
||||||
|
class UserAdminTest(BaseUserAdmin):
|
||||||
|
list_filter = (MainFactionFilter,)
|
||||||
|
|
||||||
|
my_modeladmin = UserAdminTest(User, AdminSite())
|
||||||
|
|
||||||
|
# Make sure the lookups are correct
|
||||||
|
request = self.factory.get('/')
|
||||||
|
request.user = self.user_4
|
||||||
|
changelist = my_modeladmin.get_changelist_instance(request)
|
||||||
|
filters = changelist.get_filters(request)
|
||||||
|
filterspec = filters[0][0]
|
||||||
|
expected = [
|
||||||
|
(999, 'The X-Men'),
|
||||||
|
]
|
||||||
|
self.assertEqual(filterspec.lookup_choices, expected)
|
||||||
|
|
||||||
|
# Make sure the correct queryset is returned
|
||||||
|
request = self.factory.get(
|
||||||
|
'/',
|
||||||
|
{'main_faction_id__exact': self.character_4.faction_id}
|
||||||
|
)
|
||||||
|
request.user = self.user_4
|
||||||
|
changelist = my_modeladmin.get_changelist_instance(request)
|
||||||
|
queryset = changelist.get_queryset(request)
|
||||||
|
expected = [self.user_4]
|
||||||
|
self.assertSetEqual(set(queryset), set(expected))
|
||||||
|
|
||||||
def test_change_view_loads_normally(self):
|
def test_change_view_loads_normally(self):
|
||||||
User.objects.create_superuser(
|
User.objects.create_superuser(
|
||||||
username='superuser', password='secret', email='admin@example.com'
|
username='superuser', password='secret', email='admin@example.com'
|
||||||
@@ -479,6 +543,42 @@ class TestUserAdmin(TestCaseWithTestData):
|
|||||||
self.assertEqual(response.status_code, expected)
|
self.assertEqual(response.status_code, expected)
|
||||||
|
|
||||||
|
|
||||||
|
class TestUserAdminChangeForm(TestCase):
|
||||||
|
@classmethod
|
||||||
|
def setUpClass(cls) -> None:
|
||||||
|
super().setUpClass()
|
||||||
|
cls.modeladmin = UserAdmin(model=User, admin_site=AdminSite())
|
||||||
|
|
||||||
|
def test_should_show_groups_available_to_user_with_blue_state_only(self):
|
||||||
|
# given
|
||||||
|
superuser = User.objects.create_superuser("Super")
|
||||||
|
user = AuthUtils.create_user("Bruce Wayne")
|
||||||
|
character = AuthUtils.add_main_character_2(
|
||||||
|
user,
|
||||||
|
name="Bruce Wayne",
|
||||||
|
character_id=1001,
|
||||||
|
corp_id=2001,
|
||||||
|
corp_name="Wayne Technologies"
|
||||||
|
)
|
||||||
|
blue_state = State.objects.get(name="Blue")
|
||||||
|
blue_state.member_characters.add(character)
|
||||||
|
member_state = AuthUtils.get_member_state()
|
||||||
|
group_1 = Group.objects.create(name="Group 1")
|
||||||
|
group_2 = Group.objects.create(name="Group 2")
|
||||||
|
group_2.authgroup.states.add(blue_state)
|
||||||
|
group_3 = Group.objects.create(name="Group 3")
|
||||||
|
group_3.authgroup.states.add(member_state)
|
||||||
|
self.client.force_login(superuser)
|
||||||
|
# when
|
||||||
|
response = self.client.get(f"/admin/authentication/user/{user.pk}/change/")
|
||||||
|
# then
|
||||||
|
self.assertEqual(response.status_code, 200)
|
||||||
|
soup = BeautifulSoup(response.rendered_content, features="html.parser")
|
||||||
|
groups_select = soup.find("select", {"id": "id_groups"}).find_all('option')
|
||||||
|
group_ids = {int(option["value"]) for option in groups_select}
|
||||||
|
self.assertSetEqual(group_ids, {group_1.pk, group_2.pk})
|
||||||
|
|
||||||
|
|
||||||
class TestMakeServicesHooksActions(TestCaseWithTestData):
|
class TestMakeServicesHooksActions(TestCaseWithTestData):
|
||||||
|
|
||||||
class MyServicesHookTypeA(ServicesHook):
|
class MyServicesHookTypeA(ServicesHook):
|
||||||
|
|||||||
@@ -114,12 +114,12 @@ class TestAuthenticate(TestCase):
|
|||||||
def test_authenticate_main_character(self):
|
def test_authenticate_main_character(self):
|
||||||
t = Token(character_id=self.main_character.character_id, character_owner_hash='1')
|
t = Token(character_id=self.main_character.character_id, character_owner_hash='1')
|
||||||
user = StateBackend().authenticate(token=t)
|
user = StateBackend().authenticate(token=t)
|
||||||
self.assertEquals(user, self.user)
|
self.assertEqual(user, self.user)
|
||||||
|
|
||||||
def test_authenticate_alt_character(self):
|
def test_authenticate_alt_character(self):
|
||||||
t = Token(character_id=self.alt_character.character_id, character_owner_hash='2')
|
t = Token(character_id=self.alt_character.character_id, character_owner_hash='2')
|
||||||
user = StateBackend().authenticate(token=t)
|
user = StateBackend().authenticate(token=t)
|
||||||
self.assertEquals(user, self.user)
|
self.assertEqual(user, self.user)
|
||||||
|
|
||||||
def test_authenticate_unclaimed_character(self):
|
def test_authenticate_unclaimed_character(self):
|
||||||
t = Token(character_id=self.unclaimed_character.character_id, character_name=self.unclaimed_character.character_name, character_owner_hash='3')
|
t = Token(character_id=self.unclaimed_character.character_id, character_name=self.unclaimed_character.character_name, character_owner_hash='3')
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ from django.contrib.auth.models import User
|
|||||||
from django.test import TestCase
|
from django.test import TestCase
|
||||||
|
|
||||||
from allianceauth.eveonline.models import EveCharacter, EveCorporationInfo,\
|
from allianceauth.eveonline.models import EveCharacter, EveCorporationInfo,\
|
||||||
EveAllianceInfo
|
EveAllianceInfo, EveFactionInfo
|
||||||
from allianceauth.tests.auth_utils import AuthUtils
|
from allianceauth.tests.auth_utils import AuthUtils
|
||||||
from esi.errors import IncompleteResponseError
|
from esi.errors import IncompleteResponseError
|
||||||
from esi.models import Token
|
from esi.models import Token
|
||||||
@@ -36,8 +36,8 @@ class CharacterOwnershipTestCase(TestCase):
|
|||||||
character_owner_hash='1',
|
character_owner_hash='1',
|
||||||
)
|
)
|
||||||
co = CharacterOwnership.objects.get(character=self.character)
|
co = CharacterOwnership.objects.get(character=self.character)
|
||||||
self.assertEquals(co.user, self.user)
|
self.assertEqual(co.user, self.user)
|
||||||
self.assertEquals(co.owner_hash, '1')
|
self.assertEqual(co.owner_hash, '1')
|
||||||
|
|
||||||
def test_transfer_ownership(self):
|
def test_transfer_ownership(self):
|
||||||
Token.objects.create(
|
Token.objects.create(
|
||||||
@@ -54,7 +54,7 @@ class CharacterOwnershipTestCase(TestCase):
|
|||||||
)
|
)
|
||||||
co = CharacterOwnership.objects.get(character=self.character)
|
co = CharacterOwnership.objects.get(character=self.character)
|
||||||
self.assertNotEqual(self.user, co.user)
|
self.assertNotEqual(self.user, co.user)
|
||||||
self.assertEquals(self.alt_user, co.user)
|
self.assertEqual(self.alt_user, co.user)
|
||||||
|
|
||||||
def test_clear_main_character(self):
|
def test_clear_main_character(self):
|
||||||
Token.objects.create(
|
Token.objects.create(
|
||||||
@@ -80,13 +80,15 @@ class StateTestCase(TestCase):
|
|||||||
def setUpTestData(cls):
|
def setUpTestData(cls):
|
||||||
cls.user = AuthUtils.create_user('test_user', disconnect_signals=True)
|
cls.user = AuthUtils.create_user('test_user', disconnect_signals=True)
|
||||||
AuthUtils.add_main_character(cls.user, 'Test Character', '1', corp_id='1', alliance_id='1',
|
AuthUtils.add_main_character(cls.user, 'Test Character', '1', corp_id='1', alliance_id='1',
|
||||||
corp_name='Test Corp', alliance_name='Test Alliance')
|
corp_name='Test Corp', alliance_name='Test Alliance', faction_id=1337,
|
||||||
|
faction_name='Permabanned')
|
||||||
cls.guest_state = get_guest_state()
|
cls.guest_state = get_guest_state()
|
||||||
cls.test_character = EveCharacter.objects.get(character_id='1')
|
cls.test_character = EveCharacter.objects.get(character_id='1')
|
||||||
cls.test_corporation = EveCorporationInfo.objects.create(corporation_id='1', corporation_name='Test Corp',
|
cls.test_corporation = EveCorporationInfo.objects.create(corporation_id='1', corporation_name='Test Corp',
|
||||||
corporation_ticker='TEST', member_count=1)
|
corporation_ticker='TEST', member_count=1)
|
||||||
cls.test_alliance = EveAllianceInfo.objects.create(alliance_id='1', alliance_name='Test Alliance',
|
cls.test_alliance = EveAllianceInfo.objects.create(alliance_id='1', alliance_name='Test Alliance',
|
||||||
alliance_ticker='TEST', executor_corp_id='1')
|
alliance_ticker='TEST', executor_corp_id='1')
|
||||||
|
cls.test_faction = EveFactionInfo.objects.create(faction_id=1337, faction_name='Permabanned')
|
||||||
cls.member_state = State.objects.create(
|
cls.member_state = State.objects.create(
|
||||||
name='Test Member',
|
name='Test Member',
|
||||||
priority=150,
|
priority=150,
|
||||||
@@ -98,29 +100,38 @@ class StateTestCase(TestCase):
|
|||||||
def test_state_assignment_on_character_change(self):
|
def test_state_assignment_on_character_change(self):
|
||||||
self.member_state.member_characters.add(self.test_character)
|
self.member_state.member_characters.add(self.test_character)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.user.profile.state, self.member_state)
|
self.assertEqual(self.user.profile.state, self.member_state)
|
||||||
|
|
||||||
self.member_state.member_characters.remove(self.test_character)
|
self.member_state.member_characters.remove(self.test_character)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.user.profile.state, self.guest_state)
|
self.assertEqual(self.user.profile.state, self.guest_state)
|
||||||
|
|
||||||
def test_state_assignment_on_corporation_change(self):
|
def test_state_assignment_on_corporation_change(self):
|
||||||
self.member_state.member_corporations.add(self.test_corporation)
|
self.member_state.member_corporations.add(self.test_corporation)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.user.profile.state, self.member_state)
|
self.assertEqual(self.user.profile.state, self.member_state)
|
||||||
|
|
||||||
self.member_state.member_corporations.remove(self.test_corporation)
|
self.member_state.member_corporations.remove(self.test_corporation)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.user.profile.state, self.guest_state)
|
self.assertEqual(self.user.profile.state, self.guest_state)
|
||||||
|
|
||||||
def test_state_assignment_on_alliance_addition(self):
|
def test_state_assignment_on_alliance_addition(self):
|
||||||
self.member_state.member_alliances.add(self.test_alliance)
|
self.member_state.member_alliances.add(self.test_alliance)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.user.profile.state, self.member_state)
|
self.assertEqual(self.user.profile.state, self.member_state)
|
||||||
|
|
||||||
self.member_state.member_alliances.remove(self.test_alliance)
|
self.member_state.member_alliances.remove(self.test_alliance)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.user.profile.state, self.guest_state)
|
self.assertEqual(self.user.profile.state, self.guest_state)
|
||||||
|
|
||||||
|
def test_state_assignment_on_faction_change(self):
|
||||||
|
self.member_state.member_factions.add(self.test_faction)
|
||||||
|
self._refresh_user()
|
||||||
|
self.assertEqual(self.user.profile.state, self.member_state)
|
||||||
|
|
||||||
|
self.member_state.member_factions.remove(self.test_faction)
|
||||||
|
self._refresh_user()
|
||||||
|
self.assertEqual(self.user.profile.state, self.guest_state)
|
||||||
|
|
||||||
def test_state_assignment_on_higher_priority_state_creation(self):
|
def test_state_assignment_on_higher_priority_state_creation(self):
|
||||||
self.member_state.member_characters.add(self.test_character)
|
self.member_state.member_characters.add(self.test_character)
|
||||||
@@ -130,10 +141,10 @@ class StateTestCase(TestCase):
|
|||||||
)
|
)
|
||||||
higher_state.member_characters.add(self.test_character)
|
higher_state.member_characters.add(self.test_character)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(higher_state, self.user.profile.state)
|
self.assertEqual(higher_state, self.user.profile.state)
|
||||||
higher_state.member_characters.clear()
|
higher_state.member_characters.clear()
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.member_state, self.user.profile.state)
|
self.assertEqual(self.member_state, self.user.profile.state)
|
||||||
self.member_state.member_characters.clear()
|
self.member_state.member_characters.clear()
|
||||||
|
|
||||||
def test_state_assignment_on_lower_priority_state_creation(self):
|
def test_state_assignment_on_lower_priority_state_creation(self):
|
||||||
@@ -144,10 +155,10 @@ class StateTestCase(TestCase):
|
|||||||
)
|
)
|
||||||
lower_state.member_characters.add(self.test_character)
|
lower_state.member_characters.add(self.test_character)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.member_state, self.user.profile.state)
|
self.assertEqual(self.member_state, self.user.profile.state)
|
||||||
lower_state.member_characters.clear()
|
lower_state.member_characters.clear()
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.member_state, self.user.profile.state)
|
self.assertEqual(self.member_state, self.user.profile.state)
|
||||||
self.member_state.member_characters.clear()
|
self.member_state.member_characters.clear()
|
||||||
|
|
||||||
def test_state_assignment_on_priority_change(self):
|
def test_state_assignment_on_priority_change(self):
|
||||||
@@ -161,11 +172,11 @@ class StateTestCase(TestCase):
|
|||||||
lower_state.priority = 500
|
lower_state.priority = 500
|
||||||
lower_state.save()
|
lower_state.save()
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(lower_state, self.user.profile.state)
|
self.assertEqual(lower_state, self.user.profile.state)
|
||||||
lower_state.priority = 125
|
lower_state.priority = 125
|
||||||
lower_state.save()
|
lower_state.save()
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.member_state, self.user.profile.state)
|
self.assertEqual(self.member_state, self.user.profile.state)
|
||||||
|
|
||||||
def test_state_assignment_on_state_deletion(self):
|
def test_state_assignment_on_state_deletion(self):
|
||||||
self.member_state.member_characters.add(self.test_character)
|
self.member_state.member_characters.add(self.test_character)
|
||||||
@@ -175,11 +186,11 @@ class StateTestCase(TestCase):
|
|||||||
)
|
)
|
||||||
higher_state.member_characters.add(self.test_character)
|
higher_state.member_characters.add(self.test_character)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(higher_state, self.user.profile.state)
|
self.assertEqual(higher_state, self.user.profile.state)
|
||||||
higher_state.delete()
|
higher_state.delete()
|
||||||
self.assertFalse(State.objects.filter(name='Higher State').count())
|
self.assertFalse(State.objects.filter(name='Higher State').count())
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.member_state, self.user.profile.state)
|
self.assertEqual(self.member_state, self.user.profile.state)
|
||||||
|
|
||||||
def test_state_assignment_on_public_toggle(self):
|
def test_state_assignment_on_public_toggle(self):
|
||||||
self.member_state.member_characters.add(self.test_character)
|
self.member_state.member_characters.add(self.test_character)
|
||||||
@@ -188,26 +199,26 @@ class StateTestCase(TestCase):
|
|||||||
priority=200,
|
priority=200,
|
||||||
)
|
)
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.member_state, self.user.profile.state)
|
self.assertEqual(self.member_state, self.user.profile.state)
|
||||||
higher_state.public = True
|
higher_state.public = True
|
||||||
higher_state.save()
|
higher_state.save()
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(higher_state, self.user.profile.state)
|
self.assertEqual(higher_state, self.user.profile.state)
|
||||||
higher_state.public = False
|
higher_state.public = False
|
||||||
higher_state.save()
|
higher_state.save()
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.member_state, self.user.profile.state)
|
self.assertEqual(self.member_state, self.user.profile.state)
|
||||||
|
|
||||||
def test_state_assignment_on_active_changed(self):
|
def test_state_assignment_on_active_changed(self):
|
||||||
self.member_state.member_characters.add(self.test_character)
|
self.member_state.member_characters.add(self.test_character)
|
||||||
self.user.is_active = False
|
self.user.is_active = False
|
||||||
self.user.save()
|
self.user.save()
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.user.profile.state, self.guest_state)
|
self.assertEqual(self.user.profile.state, self.guest_state)
|
||||||
self.user.is_active = True
|
self.user.is_active = True
|
||||||
self.user.save()
|
self.user.save()
|
||||||
self._refresh_user()
|
self._refresh_user()
|
||||||
self.assertEquals(self.user.profile.state, self.member_state)
|
self.assertEqual(self.user.profile.state, self.member_state)
|
||||||
|
|
||||||
|
|
||||||
class CharacterOwnershipCheckTestCase(TestCase):
|
class CharacterOwnershipCheckTestCase(TestCase):
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
from math import ceil
|
from math import ceil
|
||||||
from unittest.mock import patch
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
import requests
|
||||||
import requests_mock
|
import requests_mock
|
||||||
from packaging.version import Version as Pep440Version
|
from packaging.version import Version as Pep440Version
|
||||||
|
|
||||||
@@ -54,7 +55,6 @@ TEST_VERSION = '2.6.5'
|
|||||||
|
|
||||||
|
|
||||||
class TestStatusOverviewTag(TestCase):
|
class TestStatusOverviewTag(TestCase):
|
||||||
|
|
||||||
@patch(MODULE_PATH + '.admin_status.__version__', TEST_VERSION)
|
@patch(MODULE_PATH + '.admin_status.__version__', TEST_VERSION)
|
||||||
@patch(MODULE_PATH + '.admin_status._fetch_celery_queue_length')
|
@patch(MODULE_PATH + '.admin_status._fetch_celery_queue_length')
|
||||||
@patch(MODULE_PATH + '.admin_status._current_version_summary')
|
@patch(MODULE_PATH + '.admin_status._current_version_summary')
|
||||||
@@ -65,6 +65,7 @@ class TestStatusOverviewTag(TestCase):
|
|||||||
mock_current_version_info,
|
mock_current_version_info,
|
||||||
mock_fetch_celery_queue_length
|
mock_fetch_celery_queue_length
|
||||||
):
|
):
|
||||||
|
# given
|
||||||
notifications = {
|
notifications = {
|
||||||
'notifications': GITHUB_NOTIFICATION_ISSUES[:5]
|
'notifications': GITHUB_NOTIFICATION_ISSUES[:5]
|
||||||
}
|
}
|
||||||
@@ -82,22 +83,20 @@ class TestStatusOverviewTag(TestCase):
|
|||||||
}
|
}
|
||||||
mock_current_version_info.return_value = version_info
|
mock_current_version_info.return_value = version_info
|
||||||
mock_fetch_celery_queue_length.return_value = 3
|
mock_fetch_celery_queue_length.return_value = 3
|
||||||
|
# when
|
||||||
result = status_overview()
|
result = status_overview()
|
||||||
expected = {
|
# then
|
||||||
'notifications': GITHUB_NOTIFICATION_ISSUES[:5],
|
self.assertEqual(result["notifications"], GITHUB_NOTIFICATION_ISSUES[:5])
|
||||||
'latest_major': True,
|
self.assertTrue(result["latest_major"])
|
||||||
'latest_minor': True,
|
self.assertTrue(result["latest_minor"])
|
||||||
'latest_patch': True,
|
self.assertTrue(result["latest_patch"])
|
||||||
'latest_beta': False,
|
self.assertFalse(result["latest_beta"])
|
||||||
'current_version': TEST_VERSION,
|
self.assertEqual(result["current_version"], TEST_VERSION)
|
||||||
'latest_major_version': '2.4.5',
|
self.assertEqual(result["latest_major_version"], '2.4.5')
|
||||||
'latest_minor_version': '2.4.0',
|
self.assertEqual(result["latest_minor_version"], '2.4.0')
|
||||||
'latest_patch_version': '2.4.5',
|
self.assertEqual(result["latest_patch_version"], '2.4.5')
|
||||||
'latest_beta_version': '2.4.4a1',
|
self.assertEqual(result["latest_beta_version"], '2.4.4a1')
|
||||||
'task_queue_length': 3,
|
self.assertEqual(result["task_queue_length"], 3)
|
||||||
}
|
|
||||||
self.assertEqual(result, expected)
|
|
||||||
|
|
||||||
|
|
||||||
class TestNotifications(TestCase):
|
class TestNotifications(TestCase):
|
||||||
@@ -169,11 +168,7 @@ class TestVersionTags(TestCase):
|
|||||||
# when
|
# when
|
||||||
result = _current_version_summary()
|
result = _current_version_summary()
|
||||||
# then
|
# then
|
||||||
self.assertTrue(result['latest_major'])
|
|
||||||
self.assertTrue(result['latest_minor'])
|
|
||||||
self.assertTrue(result['latest_patch'])
|
self.assertTrue(result['latest_patch'])
|
||||||
self.assertEqual(result['latest_major_version'], '2.0.0')
|
|
||||||
self.assertEqual(result['latest_minor_version'], '2.4.0')
|
|
||||||
self.assertEqual(result['latest_patch_version'], '2.4.5')
|
self.assertEqual(result['latest_patch_version'], '2.4.5')
|
||||||
self.assertEqual(result['latest_beta_version'], '2.4.6a1')
|
self.assertEqual(result['latest_beta_version'], '2.4.6a1')
|
||||||
|
|
||||||
@@ -222,9 +217,7 @@ class TestLatestsVersion(TestCase):
|
|||||||
tags = create_tags_list(
|
tags = create_tags_list(
|
||||||
['2.1.1', '2.1.0', '2.0.0', '2.1.1a1', '1.1.1', '1.1.0', '1.0.0']
|
['2.1.1', '2.1.0', '2.0.0', '2.1.1a1', '1.1.1', '1.1.0', '1.0.0']
|
||||||
)
|
)
|
||||||
major, minor, patch, beta = _latests_versions(tags)
|
patch, beta = _latests_versions(tags)
|
||||||
self.assertEqual(major, Pep440Version('2.0.0'))
|
|
||||||
self.assertEqual(minor, Pep440Version('2.1.0'))
|
|
||||||
self.assertEqual(patch, Pep440Version('2.1.1'))
|
self.assertEqual(patch, Pep440Version('2.1.1'))
|
||||||
self.assertEqual(beta, Pep440Version('2.1.1a1'))
|
self.assertEqual(beta, Pep440Version('2.1.1a1'))
|
||||||
|
|
||||||
@@ -233,9 +226,7 @@ class TestLatestsVersion(TestCase):
|
|||||||
tags = create_tags_list(
|
tags = create_tags_list(
|
||||||
['2.1.2', '2.1.1', '2.0.1', '2.1.1a1', '1.1.1', '1.1.0', '1.0.0']
|
['2.1.2', '2.1.1', '2.0.1', '2.1.1a1', '1.1.1', '1.1.0', '1.0.0']
|
||||||
)
|
)
|
||||||
major, minor, patch, beta = _latests_versions(tags)
|
patch, beta = _latests_versions(tags)
|
||||||
self.assertEqual(major, Pep440Version('2.0.1'))
|
|
||||||
self.assertEqual(minor, Pep440Version('2.1.1'))
|
|
||||||
self.assertEqual(patch, Pep440Version('2.1.2'))
|
self.assertEqual(patch, Pep440Version('2.1.2'))
|
||||||
self.assertEqual(beta, Pep440Version('2.1.1a1'))
|
self.assertEqual(beta, Pep440Version('2.1.1a1'))
|
||||||
|
|
||||||
@@ -244,9 +235,7 @@ class TestLatestsVersion(TestCase):
|
|||||||
tags = create_tags_list(
|
tags = create_tags_list(
|
||||||
['2.1.1', '2.1.0', '2.0.0', '2.1.1a1', 'invalid']
|
['2.1.1', '2.1.0', '2.0.0', '2.1.1a1', 'invalid']
|
||||||
)
|
)
|
||||||
major, minor, patch, beta = _latests_versions(tags)
|
patch, beta = _latests_versions(tags)
|
||||||
self.assertEqual(major, Pep440Version('2.0.0'))
|
|
||||||
self.assertEqual(minor, Pep440Version('2.1.0'))
|
|
||||||
self.assertEqual(patch, Pep440Version('2.1.1'))
|
self.assertEqual(patch, Pep440Version('2.1.1'))
|
||||||
self.assertEqual(beta, Pep440Version('2.1.1a1'))
|
self.assertEqual(beta, Pep440Version('2.1.1a1'))
|
||||||
|
|
||||||
@@ -317,3 +306,25 @@ class TestFetchListFromGitlab(TestCase):
|
|||||||
result = _fetch_list_from_gitlab(self.url, max_pages=max_pages)
|
result = _fetch_list_from_gitlab(self.url, max_pages=max_pages)
|
||||||
self.assertEqual(result, GITHUB_TAGS[:4])
|
self.assertEqual(result, GITHUB_TAGS[:4])
|
||||||
self.assertEqual(requests_mocker.call_count, max_pages)
|
self.assertEqual(requests_mocker.call_count, max_pages)
|
||||||
|
|
||||||
|
@requests_mock.mock()
|
||||||
|
@patch(MODULE_PATH + '.admin_status.logger')
|
||||||
|
def test_should_not_raise_any_exception_from_github_request_but_log_as_warning(
|
||||||
|
self, requests_mocker, mock_logger
|
||||||
|
):
|
||||||
|
for my_exception in [
|
||||||
|
requests.exceptions.ConnectionError,
|
||||||
|
requests.exceptions.HTTPError,
|
||||||
|
requests.exceptions.URLRequired,
|
||||||
|
requests.exceptions.TooManyRedirects,
|
||||||
|
requests.exceptions.ConnectTimeout,
|
||||||
|
requests.exceptions.Timeout,
|
||||||
|
|
||||||
|
]:
|
||||||
|
requests_mocker.get(self.url, exc=my_exception)
|
||||||
|
try:
|
||||||
|
result = _fetch_list_from_gitlab(self.url)
|
||||||
|
except Exception as ex:
|
||||||
|
self.fail(f"Unexpected exception raised: {ex}")
|
||||||
|
self.assertTrue(mock_logger.warning.called)
|
||||||
|
self.assertListEqual(result, [])
|
||||||
|
|||||||
@@ -6,8 +6,10 @@ from django.contrib.auth import login, authenticate
|
|||||||
from django.contrib.auth.decorators import login_required
|
from django.contrib.auth.decorators import login_required
|
||||||
from django.contrib.auth.models import User
|
from django.contrib.auth.models import User
|
||||||
from django.core import signing
|
from django.core import signing
|
||||||
|
from django.core.mail import EmailMultiAlternatives
|
||||||
from django.http import JsonResponse
|
from django.http import JsonResponse
|
||||||
from django.shortcuts import redirect, render
|
from django.shortcuts import redirect, render
|
||||||
|
from django.template.loader import render_to_string
|
||||||
from django.urls import reverse, reverse_lazy
|
from django.urls import reverse, reverse_lazy
|
||||||
from django.utils.translation import gettext_lazy as _
|
from django.utils.translation import gettext_lazy as _
|
||||||
|
|
||||||
@@ -63,7 +65,7 @@ def dashboard(request):
|
|||||||
@login_required
|
@login_required
|
||||||
@token_required(scopes=settings.LOGIN_TOKEN_SCOPES)
|
@token_required(scopes=settings.LOGIN_TOKEN_SCOPES)
|
||||||
def main_character_change(request, token):
|
def main_character_change(request, token):
|
||||||
logger.debug("main_character_change called by user %s for character %s" % (request.user, token.character_name))
|
logger.debug(f"main_character_change called by user {request.user} for character {token.character_name}")
|
||||||
try:
|
try:
|
||||||
co = CharacterOwnership.objects.get(character__character_id=token.character_id, user=request.user)
|
co = CharacterOwnership.objects.get(character__character_id=token.character_id, user=request.user)
|
||||||
except CharacterOwnership.DoesNotExist:
|
except CharacterOwnership.DoesNotExist:
|
||||||
@@ -137,9 +139,52 @@ class RegistrationView(BaseRegistrationView):
|
|||||||
form_class = RegistrationForm
|
form_class = RegistrationForm
|
||||||
template_name = "public/register.html"
|
template_name = "public/register.html"
|
||||||
email_body_template = "registration/activation_email.txt"
|
email_body_template = "registration/activation_email.txt"
|
||||||
|
email_body_template_html = "registration/activation_email_html.txt"
|
||||||
email_subject_template = "registration/activation_email_subject.txt"
|
email_subject_template = "registration/activation_email_subject.txt"
|
||||||
success_url = reverse_lazy('registration_complete')
|
success_url = reverse_lazy('registration_complete')
|
||||||
|
|
||||||
|
def send_activation_email(self, user):
|
||||||
|
"""
|
||||||
|
Implement our own way to send a mail to make sure we
|
||||||
|
send a RFC conform multipart email
|
||||||
|
:param user:
|
||||||
|
:type user:
|
||||||
|
"""
|
||||||
|
|
||||||
|
activation_key = self.get_activation_key(user)
|
||||||
|
context = self.get_email_context(activation_key)
|
||||||
|
context["user"] = user
|
||||||
|
|
||||||
|
# email subject
|
||||||
|
subject = render_to_string(
|
||||||
|
template_name=self.email_subject_template,
|
||||||
|
context=context,
|
||||||
|
request=self.request,
|
||||||
|
)
|
||||||
|
subject = "".join(subject.splitlines())
|
||||||
|
|
||||||
|
# plaintext email body part
|
||||||
|
message = render_to_string(
|
||||||
|
template_name=self.email_body_template,
|
||||||
|
context=context,
|
||||||
|
request=self.request,
|
||||||
|
)
|
||||||
|
|
||||||
|
# html email body part
|
||||||
|
message_html = render_to_string(
|
||||||
|
template_name=self.email_body_template_html,
|
||||||
|
context=context,
|
||||||
|
request=self.request,
|
||||||
|
)
|
||||||
|
|
||||||
|
# send it
|
||||||
|
user.email_user(
|
||||||
|
subject,
|
||||||
|
message,
|
||||||
|
settings.DEFAULT_FROM_EMAIL,
|
||||||
|
**{'html_message': message_html},
|
||||||
|
)
|
||||||
|
|
||||||
def get_success_url(self, user):
|
def get_success_url(self, user):
|
||||||
if not getattr(settings, 'REGISTRATION_VERIFY_EMAIL', True):
|
if not getattr(settings, 'REGISTRATION_VERIFY_EMAIL', True):
|
||||||
return reverse_lazy('authentication:dashboard')
|
return reverse_lazy('authentication:dashboard')
|
||||||
@@ -154,7 +199,7 @@ class RegistrationView(BaseRegistrationView):
|
|||||||
if not getattr(settings, 'REGISTRATION_VERIFY_EMAIL', True):
|
if not getattr(settings, 'REGISTRATION_VERIFY_EMAIL', True):
|
||||||
# Keep the request so the user can be automagically logged in.
|
# Keep the request so the user can be automagically logged in.
|
||||||
setattr(self, 'request', request)
|
setattr(self, 'request', request)
|
||||||
return super(RegistrationView, self).dispatch(request, *args, **kwargs)
|
return super().dispatch(request, *args, **kwargs)
|
||||||
|
|
||||||
def register(self, form):
|
def register(self, form):
|
||||||
user = User.objects.get(pk=self.request.session.get('registration_uid'))
|
user = User.objects.get(pk=self.request.session.get('registration_uid'))
|
||||||
@@ -173,7 +218,7 @@ class RegistrationView(BaseRegistrationView):
|
|||||||
return signing.dumps(obj=[getattr(user, User.USERNAME_FIELD), user.email], salt=REGISTRATION_SALT)
|
return signing.dumps(obj=[getattr(user, User.USERNAME_FIELD), user.email], salt=REGISTRATION_SALT)
|
||||||
|
|
||||||
def get_email_context(self, activation_key):
|
def get_email_context(self, activation_key):
|
||||||
context = super(RegistrationView, self).get_email_context(activation_key)
|
context = super().get_email_context(activation_key)
|
||||||
context['url'] = context['site'].domain + reverse('registration_activate', args=[activation_key])
|
context['url'] = context['site'].domain + reverse('registration_activate', args=[activation_key])
|
||||||
return context
|
return context
|
||||||
|
|
||||||
|
|||||||
@@ -41,7 +41,7 @@ def create_project(parser, options, args):
|
|||||||
# Call the command with extra context
|
# Call the command with extra context
|
||||||
call_command(StartProject(), *args, **command_options)
|
call_command(StartProject(), *args, **command_options)
|
||||||
|
|
||||||
print("Success! %(project_name)s has been created." % {'project_name': args[0]}) # noqa
|
print(f"Success! {args[0]} has been created.") # noqa
|
||||||
|
|
||||||
|
|
||||||
def update_settings(parser, options, args):
|
def update_settings(parser, options, args):
|
||||||
@@ -69,10 +69,10 @@ def update_settings(parser, options, args):
|
|||||||
template_settings_path = os.path.join(template_path, 'project_name/settings/base.py')
|
template_settings_path = os.path.join(template_path, 'project_name/settings/base.py')
|
||||||
|
|
||||||
# overwrite the local project's base settings
|
# overwrite the local project's base settings
|
||||||
with open(template_settings_path, 'r') as template, open(settings_path, 'w') as target:
|
with open(template_settings_path) as template, open(settings_path, 'w') as target:
|
||||||
target.write(template.read())
|
target.write(template.read())
|
||||||
|
|
||||||
print("Successfully updated %(project_name)s settings." % {'project_name': project_name})
|
print(f"Successfully updated {project_name} settings.")
|
||||||
|
|
||||||
|
|
||||||
COMMANDS = {
|
COMMANDS = {
|
||||||
|
|||||||
@@ -29,7 +29,7 @@ class CorpStatsQuerySet(models.QuerySet):
|
|||||||
if user.has_perm('corputils.view_state_corpstats'):
|
if user.has_perm('corputils.view_state_corpstats'):
|
||||||
queries.append(models.Q(corp__in=user.profile.state.member_corporations.all()))
|
queries.append(models.Q(corp__in=user.profile.state.member_corporations.all()))
|
||||||
queries.append(models.Q(corp__alliance__in=user.profile.state.member_alliances.all()))
|
queries.append(models.Q(corp__alliance__in=user.profile.state.member_alliances.all()))
|
||||||
logger.debug('%s queries for user %s visible corpstats.' % (len(queries), user))
|
logger.debug(f'{len(queries)} queries for user {user} visible corpstats.')
|
||||||
# filter based on queries
|
# filter based on queries
|
||||||
query = queries.pop()
|
query = queries.pop()
|
||||||
for q in queries:
|
for q in queries:
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-12-14 21:36
|
# Generated by Django 1.10.1 on 2016-12-14 21:36
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
import django.db.models.deletion
|
import django.db.models.deletion
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-12-14 21:48
|
# Generated by Django 1.10.1 on 2016-12-14 21:48
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.5 on 2017-03-22 23:35
|
# Generated by Django 1.10.5 on 2017-03-22 23:35
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.5 on 2017-03-26 20:13
|
# Generated by Django 1.10.5 on 2017-03-26 20:13
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
import django.db.models.deletion
|
import django.db.models.deletion
|
||||||
@@ -48,7 +47,7 @@ class Migration(migrations.Migration):
|
|||||||
),
|
),
|
||||||
migrations.AlterUniqueTogether(
|
migrations.AlterUniqueTogether(
|
||||||
name='corpmember',
|
name='corpmember',
|
||||||
unique_together=set([('corpstats', 'character_id')]),
|
unique_together={('corpstats', 'character_id')},
|
||||||
),
|
),
|
||||||
migrations.RunPython(convert_json_to_members, convert_members_to_json),
|
migrations.RunPython(convert_json_to_members, convert_members_to_json),
|
||||||
migrations.RemoveField(
|
migrations.RemoveField(
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.11.2 on 2017-06-10 15:34
|
# Generated by Django 1.11.2 on 2017-06-10 15:34
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ class CorpStats(models.Model):
|
|||||||
objects = CorpStatsManager()
|
objects = CorpStatsManager()
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return "%s for %s" % (self.__class__.__name__, self.corp)
|
return f"{self.__class__.__name__} for {self.corp}"
|
||||||
|
|
||||||
def update(self):
|
def update(self):
|
||||||
try:
|
try:
|
||||||
@@ -73,7 +73,7 @@ class CorpStats(models.Model):
|
|||||||
self.save()
|
self.save()
|
||||||
|
|
||||||
except TokenError as e:
|
except TokenError as e:
|
||||||
logger.warning("%s failed to update: %s" % (self, e))
|
logger.warning(f"{self} failed to update: {e}")
|
||||||
if self.token.user:
|
if self.token.user:
|
||||||
notify(
|
notify(
|
||||||
self.token.user, "%s failed to update with your ESI token." % self,
|
self.token.user, "%s failed to update with your ESI token." % self,
|
||||||
@@ -81,9 +81,9 @@ class CorpStats(models.Model):
|
|||||||
level="error")
|
level="error")
|
||||||
self.delete()
|
self.delete()
|
||||||
except HTTPForbidden as e:
|
except HTTPForbidden as e:
|
||||||
logger.warning("%s failed to update: %s" % (self, e))
|
logger.warning(f"{self} failed to update: {e}")
|
||||||
if self.token.user:
|
if self.token.user:
|
||||||
notify(self.token.user, "%s failed to update with your ESI token." % self, message="%s: %s" % (e.status_code, e.message), level="error")
|
notify(self.token.user, "%s failed to update with your ESI token." % self, message=f"{e.status_code}: {e.message}", level="error")
|
||||||
self.delete()
|
self.delete()
|
||||||
except AssertionError:
|
except AssertionError:
|
||||||
logger.warning("%s token character no longer in corp." % self)
|
logger.warning("%s token character no longer in corp." % self)
|
||||||
@@ -99,7 +99,7 @@ class CorpStats(models.Model):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def user_count(self):
|
def user_count(self):
|
||||||
return len(set([m.main_character for m in self.members.all() if m.main_character]))
|
return len({m.main_character for m in self.members.all() if m.main_character})
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def registered_member_count(self):
|
def registered_member_count(self):
|
||||||
|
|||||||
@@ -1,15 +1,15 @@
|
|||||||
{% extends 'allianceauth/base.html' %}
|
{% extends 'allianceauth/base.html' %}
|
||||||
{% load i18n %}
|
{% load i18n %}
|
||||||
{% block page_title %}{% trans "Corporation Member Data" %}{% endblock %}
|
{% block page_title %}{% translate "Corporation Member Data" %}{% endblock %}
|
||||||
{% block content %}
|
{% block content %}
|
||||||
<div class="col-lg-12">
|
<div class="col-lg-12">
|
||||||
<h1 class="page-header text-center">{% trans "Corporation Member Data" %}</h1>
|
<h1 class="page-header text-center">{% translate "Corporation Member Data" %}</h1>
|
||||||
<div class="col-lg-10 col-lg-offset-1 container">
|
<div class="col-lg-10 col-lg-offset-1 container">
|
||||||
<nav class="navbar navbar-default">
|
<nav class="navbar navbar-default">
|
||||||
<div class="container-fluid">
|
<div class="container-fluid">
|
||||||
<ul class="nav navbar-nav">
|
<ul class="nav navbar-nav">
|
||||||
<li class="dropdown">
|
<li class="dropdown">
|
||||||
<a href="#" id="dLabel" class="dropdown-toggle" role="button" data-toggle="dropdown" aria-haspopup="false" aria-expanded="false">{% trans "Corporations" %}<span class="caret"></span></a>
|
<a href="#" id="dLabel" class="dropdown-toggle" role="button" data-toggle="dropdown" aria-haspopup="false" aria-expanded="false">{% translate "Corporations" %}<span class="caret"></span></a>
|
||||||
<ul class="dropdown-menu" role="menu" aria-labelledby="dLabel">
|
<ul class="dropdown-menu" role="menu" aria-labelledby="dLabel">
|
||||||
{% for corpstat in available %}
|
{% for corpstat in available %}
|
||||||
<li>
|
<li>
|
||||||
@@ -20,13 +20,13 @@
|
|||||||
</li>
|
</li>
|
||||||
{% if perms.corputils.add_corpstats %}
|
{% if perms.corputils.add_corpstats %}
|
||||||
<li>
|
<li>
|
||||||
<a href="{% url 'corputils:add' %}">{% trans "Add" %}</a>
|
<a href="{% url 'corputils:add' %}">{% translate "Add" %}</a>
|
||||||
</li>
|
</li>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
</ul>
|
</ul>
|
||||||
<form class="navbar-form navbar-right" role="search" action="{% url 'corputils:search' %}" method="GET">
|
<form class="navbar-form navbar-right" role="search" action="{% url 'corputils:search' %}" method="GET">
|
||||||
<div class="form-group">
|
<div class="form-group">
|
||||||
<input type="text" class="form-control" name="search_string" placeholder="{% if search_string %}{{ search_string }}{% else %}{% trans "Search all corporations..." %}{% endif %}">
|
<input type="text" class="form-control" name="search_string" placeholder="{% if search_string %}{{ search_string }}{% else %}{% translate "Search all corporations..." %}{% endif %}">
|
||||||
</div>
|
</div>
|
||||||
</form>
|
</form>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
@@ -30,12 +30,12 @@
|
|||||||
<div class="panel panel-default">
|
<div class="panel panel-default">
|
||||||
<div class="panel-heading">
|
<div class="panel-heading">
|
||||||
<ul class="nav nav-pills pull-left">
|
<ul class="nav nav-pills pull-left">
|
||||||
<li class="active"><a href="#mains" data-toggle="pill">{% trans 'Mains' %} ({{ total_mains }})</a></li>
|
<li class="active"><a href="#mains" data-toggle="pill">{% translate 'Mains' %} ({{ total_mains }})</a></li>
|
||||||
<li><a href="#members" data-toggle="pill">{% trans 'Members' %} ({{ corpstats.member_count }})</a></li>
|
<li><a href="#members" data-toggle="pill">{% translate 'Members' %} ({{ corpstats.member_count }})</a></li>
|
||||||
<li><a href="#unregistered" data-toggle="pill">{% trans 'Unregistered' %} ({{ unregistered.count }})</a></li>
|
<li><a href="#unregistered" data-toggle="pill">{% translate 'Unregistered' %} ({{ unregistered.count }})</a></li>
|
||||||
</ul>
|
</ul>
|
||||||
<div class="pull-right hidden-xs">
|
<div class="pull-right hidden-xs">
|
||||||
{% trans "Last update:" %} {{ corpstats.last_update|naturaltime }}
|
{% translate "Last update:" %} {{ corpstats.last_update|naturaltime }}
|
||||||
<a class="btn btn-success" type="button" href="{% url 'corputils:update' corpstats.corp.corporation_id %}" title="Update Now">
|
<a class="btn btn-success" type="button" href="{% url 'corputils:update' corpstats.corp.corporation_id %}" title="Update Now">
|
||||||
<span class="glyphicon glyphicon-refresh"></span>
|
<span class="glyphicon glyphicon-refresh"></span>
|
||||||
</a>
|
</a>
|
||||||
@@ -71,9 +71,9 @@
|
|||||||
{% if forloop.first %}
|
{% if forloop.first %}
|
||||||
<tr>
|
<tr>
|
||||||
<th></th>
|
<th></th>
|
||||||
<th class="text-center">{% trans "Character" %}</th>
|
<th class="text-center">{% translate "Character" %}</th>
|
||||||
<th class="text-center">{% trans "Corporation" %}</th>
|
<th class="text-center">{% translate "Corporation" %}</th>
|
||||||
<th class="text-center">{% trans "Alliance" %}</th>
|
<th class="text-center">{% translate "Alliance" %}</th>
|
||||||
<th class="text-center"></th>
|
<th class="text-center"></th>
|
||||||
</tr>
|
</tr>
|
||||||
{% endif %}
|
{% endif %}
|
||||||
@@ -88,7 +88,7 @@
|
|||||||
<td class="text-center" style="width:30%">{{ alt.alliance_name }}</td>
|
<td class="text-center" style="width:30%">{{ alt.alliance_name }}</td>
|
||||||
<td class="text-center" style="width:5%">
|
<td class="text-center" style="width:5%">
|
||||||
<a href="https://zkillboard.com/character/{{ alt.character_id }}/" class="label label-danger" target="_blank">
|
<a href="https://zkillboard.com/character/{{ alt.character_id }}/" class="label label-danger" target="_blank">
|
||||||
{% trans "Killboard" %}
|
{% translate "Killboard" %}
|
||||||
</a>
|
</a>
|
||||||
</td>
|
</td>
|
||||||
</tr>
|
</tr>
|
||||||
@@ -109,11 +109,11 @@
|
|||||||
<thead>
|
<thead>
|
||||||
<tr>
|
<tr>
|
||||||
<th></th>
|
<th></th>
|
||||||
<th class="text-center">{% trans "Character" %}</th>
|
<th class="text-center">{% translate "Character" %}</th>
|
||||||
<th class="text-center"></th>
|
<th class="text-center"></th>
|
||||||
<th class="text-center">{% trans "Main Character" %}</th>
|
<th class="text-center">{% translate "Main Character" %}</th>
|
||||||
<th class="text-center">{% trans "Main Corporation" %}</th>
|
<th class="text-center">{% translate "Main Corporation" %}</th>
|
||||||
<th class="text-center">{% trans "Main Alliance" %}</th>
|
<th class="text-center">{% translate "Main Alliance" %}</th>
|
||||||
</tr>
|
</tr>
|
||||||
</thead>
|
</thead>
|
||||||
<tbody>
|
<tbody>
|
||||||
@@ -122,7 +122,7 @@
|
|||||||
<td><img src="{{ member.portrait_url }}" class="img-circle"></td>
|
<td><img src="{{ member.portrait_url }}" class="img-circle"></td>
|
||||||
<td class="text-center">{{ member }}</td>
|
<td class="text-center">{{ member }}</td>
|
||||||
<td class="text-center">
|
<td class="text-center">
|
||||||
<a href="https://zkillboard.com/character/{{ member.character_id }}/" class="label label-danger" target="_blank">{% trans "Killboard" %}</a>
|
<a href="https://zkillboard.com/character/{{ member.character_id }}/" class="label label-danger" target="_blank">{% translate "Killboard" %}</a>
|
||||||
</td>
|
</td>
|
||||||
<td class="text-center">{{ member.character_ownership.user.profile.main_character.character_name }}</td>
|
<td class="text-center">{{ member.character_ownership.user.profile.main_character.character_name }}</td>
|
||||||
<td class="text-center">{{ member.character_ownership.user.profile.main_character.corporation_name }}</td>
|
<td class="text-center">{{ member.character_ownership.user.profile.main_character.corporation_name }}</td>
|
||||||
@@ -134,7 +134,7 @@
|
|||||||
<td><img src="{{ member.portrait_url }}" class="img-circle"></td>
|
<td><img src="{{ member.portrait_url }}" class="img-circle"></td>
|
||||||
<td class="text-center">{{ member.character_name }}</td>
|
<td class="text-center">{{ member.character_name }}</td>
|
||||||
<td class="text-center">
|
<td class="text-center">
|
||||||
<a href="https://zkillboard.com/character/{{ member.character_id }}/" class="label label-danger" target="_blank">{% trans "Killboard" %}</a>
|
<a href="https://zkillboard.com/character/{{ member.character_id }}/" class="label label-danger" target="_blank">{% translate "Killboard" %}</a>
|
||||||
</td>
|
</td>
|
||||||
<td class="text-center"></td>
|
<td class="text-center"></td>
|
||||||
<td class="text-center"></td>
|
<td class="text-center"></td>
|
||||||
@@ -153,7 +153,7 @@
|
|||||||
<thead>
|
<thead>
|
||||||
<tr>
|
<tr>
|
||||||
<th></th>
|
<th></th>
|
||||||
<th class="text-center">{% trans "Character" %}</th>
|
<th class="text-center">{% translate "Character" %}</th>
|
||||||
<th class="text-center"></th>
|
<th class="text-center"></th>
|
||||||
</tr>
|
</tr>
|
||||||
</thead>
|
</thead>
|
||||||
@@ -164,7 +164,7 @@
|
|||||||
<td class="text-center">{{ member.character_name }}</td>
|
<td class="text-center">{{ member.character_name }}</td>
|
||||||
<td class="text-center">
|
<td class="text-center">
|
||||||
<a href="https://zkillboard.com/character/{{ member.character_id }}/" class="label label-danger" target="_blank">
|
<a href="https://zkillboard.com/character/{{ member.character_id }}/" class="label label-danger" target="_blank">
|
||||||
{% trans "Killboard" %}
|
{% translate "Killboard" %}
|
||||||
</a>
|
</a>
|
||||||
</td>
|
</td>
|
||||||
</tr>
|
</tr>
|
||||||
@@ -193,6 +193,8 @@
|
|||||||
"columnDefs": [
|
"columnDefs": [
|
||||||
{ "sortable": false, "targets": [1] },
|
{ "sortable": false, "targets": [1] },
|
||||||
],
|
],
|
||||||
|
"stateSave": true,
|
||||||
|
"stateDuration": 0
|
||||||
});
|
});
|
||||||
$('#table-members').DataTable({
|
$('#table-members').DataTable({
|
||||||
"columnDefs": [
|
"columnDefs": [
|
||||||
@@ -200,6 +202,8 @@
|
|||||||
{ "sortable": false, "targets": [0, 2] },
|
{ "sortable": false, "targets": [0, 2] },
|
||||||
],
|
],
|
||||||
"order": [[ 1, "asc" ]],
|
"order": [[ 1, "asc" ]],
|
||||||
|
"stateSave": true,
|
||||||
|
"stateDuration": 0
|
||||||
});
|
});
|
||||||
$('#table-unregistered').DataTable({
|
$('#table-unregistered').DataTable({
|
||||||
"columnDefs": [
|
"columnDefs": [
|
||||||
@@ -207,6 +211,8 @@
|
|||||||
{ "sortable": false, "targets": [0, 2] },
|
{ "sortable": false, "targets": [0, 2] },
|
||||||
],
|
],
|
||||||
"order": [[ 1, "asc" ]],
|
"order": [[ 1, "asc" ]],
|
||||||
|
"stateSave": true,
|
||||||
|
"stateDuration": 0
|
||||||
});
|
});
|
||||||
|
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -3,19 +3,19 @@
|
|||||||
{% block member_data %}
|
{% block member_data %}
|
||||||
<div class="panel panel-default">
|
<div class="panel panel-default">
|
||||||
<div class="panel-heading clearfix">
|
<div class="panel-heading clearfix">
|
||||||
<div class="panel-title pull-left">{% trans "Search Results" %}</div>
|
<div class="panel-title pull-left">{% translate "Search Results" %}</div>
|
||||||
</div>
|
</div>
|
||||||
<div class="panel-body">
|
<div class="panel-body">
|
||||||
<table class="table table-hover" id="table-search">
|
<table class="table table-hover" id="table-search">
|
||||||
<thead>
|
<thead>
|
||||||
<tr>
|
<tr>
|
||||||
<th class="text-center"></th>
|
<th class="text-center"></th>
|
||||||
<th class="text-center">{% trans "Character" %}</th>
|
<th class="text-center">{% translate "Character" %}</th>
|
||||||
<th class="text-center">{% trans "Corporation" %}</th>
|
<th class="text-center">{% translate "Corporation" %}</th>
|
||||||
<th class="text-center">{% trans "zKillboard" %}</th>
|
<th class="text-center">{% translate "zKillboard" %}</th>
|
||||||
<th class="text-center">{% trans "Main Character" %}</th>
|
<th class="text-center">{% translate "Main Character" %}</th>
|
||||||
<th class="text-center">{% trans "Main Corporation" %}</th>
|
<th class="text-center">{% translate "Main Corporation" %}</th>
|
||||||
<th class="text-center">{% trans "Main Alliance" %}</th>
|
<th class="text-center">{% translate "Main Alliance" %}</th>
|
||||||
</tr>
|
</tr>
|
||||||
</thead>
|
</thead>
|
||||||
<tbody>
|
<tbody>
|
||||||
@@ -24,7 +24,7 @@
|
|||||||
<td class="text-center"><img src="{{ result.1.portrait_url }}" class="img-circle"></td>
|
<td class="text-center"><img src="{{ result.1.portrait_url }}" class="img-circle"></td>
|
||||||
<td class="text-center">{{ result.1.character_name }}</td>
|
<td class="text-center">{{ result.1.character_name }}</td>
|
||||||
<td class="text-center">{{ result.0.corp.corporation_name }}</td>
|
<td class="text-center">{{ result.0.corp.corporation_name }}</td>
|
||||||
<td class="text-center"><a href="https://zkillboard.com/character/{{ result.1.character_id }}/" class="label label-danger" target="_blank">{% trans "Killboard" %}</a></td>
|
<td class="text-center"><a href="https://zkillboard.com/character/{{ result.1.character_id }}/" class="label label-danger" target="_blank">{% translate "Killboard" %}</a></td>
|
||||||
<td class="text-center">{{ result.1.main_character.character_name }}</td>
|
<td class="text-center">{{ result.1.main_character.character_name }}</td>
|
||||||
<td class="text-center">{{ result.1.main_character.corporation_name }}</td>
|
<td class="text-center">{{ result.1.main_character.corporation_name }}</td>
|
||||||
<td class="text-center">{{ result.1.main_character.alliance_name }}</td>
|
<td class="text-center">{{ result.1.main_character.alliance_name }}</td>
|
||||||
@@ -43,6 +43,9 @@
|
|||||||
{% endblock %}
|
{% endblock %}
|
||||||
{% block extra_script %}
|
{% block extra_script %}
|
||||||
$(document).ready(function(){
|
$(document).ready(function(){
|
||||||
$('#table-search').DataTable();
|
$('#table-search').DataTable({
|
||||||
|
"stateSave": true,
|
||||||
|
"stateDuration": 0
|
||||||
|
});
|
||||||
});
|
});
|
||||||
{% endblock %}
|
{% endblock %}
|
||||||
|
|||||||
@@ -243,7 +243,7 @@ class CorpMemberTestCase(TestCase):
|
|||||||
CharacterOwnership.objects.create(character=character, user=self.user, owner_hash='b')
|
CharacterOwnership.objects.create(character=character, user=self.user, owner_hash='b')
|
||||||
self.member.refresh_from_db()
|
self.member.refresh_from_db()
|
||||||
self.assertNotEqual(self.member.main_character, self.member.character)
|
self.assertNotEqual(self.member.main_character, self.member.character)
|
||||||
self.assertEquals(self.member.main_character, self.user.profile.main_character)
|
self.assertEqual(self.member.main_character, self.user.profile.main_character)
|
||||||
|
|
||||||
# test when is main
|
# test when is main
|
||||||
old_main = self.user.profile.main_character
|
old_main = self.user.profile.main_character
|
||||||
@@ -274,7 +274,7 @@ class CorpMemberTestCase(TestCase):
|
|||||||
AuthUtils.connect_signals()
|
AuthUtils.connect_signals()
|
||||||
|
|
||||||
def test_portrait_url(self):
|
def test_portrait_url(self):
|
||||||
self.assertEquals(self.member.portrait_url(size=32), 'https://images.evetech.net/characters/2/portrait?size=32')
|
self.assertEqual(self.member.portrait_url(size=32), 'https://images.evetech.net/characters/2/portrait?size=32')
|
||||||
self.assertEquals(self.member.portrait_url(size=32), self.member.portrait_url_32)
|
self.assertEqual(self.member.portrait_url(size=32), self.member.portrait_url_32)
|
||||||
self.assertEquals(self.member.portrait_url(size=64), self.member.portrait_url_64)
|
self.assertEqual(self.member.portrait_url(size=64), self.member.portrait_url_64)
|
||||||
self.assertEquals(self.member.portrait_url(size=128), self.member.portrait_url_128)
|
self.assertEqual(self.member.portrait_url(size=128), self.member.portrait_url_128)
|
||||||
|
|||||||
@@ -6,18 +6,19 @@ from .providers import ObjectNotFound
|
|||||||
from .models import EveAllianceInfo
|
from .models import EveAllianceInfo
|
||||||
from .models import EveCharacter
|
from .models import EveCharacter
|
||||||
from .models import EveCorporationInfo
|
from .models import EveCorporationInfo
|
||||||
|
from .models import EveFactionInfo
|
||||||
|
|
||||||
|
|
||||||
class EveEntityExistsError(forms.ValidationError):
|
class EveEntityExistsError(forms.ValidationError):
|
||||||
def __init__(self, entity_type_name, id):
|
def __init__(self, entity_type_name, id):
|
||||||
super(EveEntityExistsError, self).__init__(
|
super().__init__(
|
||||||
message='{} with ID {} already exists.'.format(entity_type_name, id))
|
message=f'{entity_type_name} with ID {id} already exists.')
|
||||||
|
|
||||||
|
|
||||||
class EveEntityNotFoundError(forms.ValidationError):
|
class EveEntityNotFoundError(forms.ValidationError):
|
||||||
def __init__(self, entity_type_name, id):
|
def __init__(self, entity_type_name, id):
|
||||||
super(EveEntityNotFoundError, self).__init__(
|
super().__init__(
|
||||||
message='{} with ID {} not found.'.format(entity_type_name, id))
|
message=f'{entity_type_name} with ID {id} not found.')
|
||||||
|
|
||||||
|
|
||||||
class EveEntityForm(forms.ModelForm):
|
class EveEntityForm(forms.ModelForm):
|
||||||
@@ -34,6 +35,38 @@ class EveEntityForm(forms.ModelForm):
|
|||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def get_faction_choices():
|
||||||
|
# use a method to avoid making an ESI call when the app loads
|
||||||
|
# restrict to only those factions a player can join for faction warfare
|
||||||
|
player_factions = [x for x in EveFactionInfo.provider.get_all_factions() if x['militia_corporation_id']]
|
||||||
|
return [(x['faction_id'], x['name']) for x in player_factions]
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class EveFactionForm(EveEntityForm):
|
||||||
|
id = forms.ChoiceField(
|
||||||
|
choices=get_faction_choices,
|
||||||
|
label="Name"
|
||||||
|
)
|
||||||
|
|
||||||
|
def clean_id(self):
|
||||||
|
try:
|
||||||
|
assert self.Meta.model.provider.get_faction(self.cleaned_data['id'])
|
||||||
|
except (AssertionError, ObjectNotFound):
|
||||||
|
raise EveEntityNotFoundError('faction', self.cleaned_data['id'])
|
||||||
|
if self.Meta.model.objects.filter(faction_id=self.cleaned_data['id']).exists():
|
||||||
|
raise EveEntityExistsError('faction', self.cleaned_data['id'])
|
||||||
|
return self.cleaned_data['id']
|
||||||
|
|
||||||
|
def save(self, commit=True):
|
||||||
|
faction = self.Meta.model.provider.get_faction(self.cleaned_data['id'])
|
||||||
|
return self.Meta.model.objects.create(faction_id=faction.id, faction_name=faction.name)
|
||||||
|
|
||||||
|
class Meta:
|
||||||
|
fields = ['id']
|
||||||
|
model = EveFactionInfo
|
||||||
|
|
||||||
|
|
||||||
class EveCharacterForm(EveEntityForm):
|
class EveCharacterForm(EveEntityForm):
|
||||||
entity_type_name = 'character'
|
entity_type_name = 'character'
|
||||||
|
|
||||||
@@ -94,6 +127,21 @@ class EveAllianceForm(EveEntityForm):
|
|||||||
model = EveAllianceInfo
|
model = EveAllianceInfo
|
||||||
|
|
||||||
|
|
||||||
|
@admin.register(EveFactionInfo)
|
||||||
|
class EveFactionInfoAdmin(admin.ModelAdmin):
|
||||||
|
search_fields = ['faction_name']
|
||||||
|
list_display = ('faction_name',)
|
||||||
|
ordering = ('faction_name',)
|
||||||
|
|
||||||
|
def has_change_permission(self, request, obj=None):
|
||||||
|
return False
|
||||||
|
|
||||||
|
def get_form(self, request, obj=None, **kwargs):
|
||||||
|
if not obj or not obj.pk:
|
||||||
|
return EveFactionForm
|
||||||
|
return super().get_form(request, obj=obj, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
@admin.register(EveCorporationInfo)
|
@admin.register(EveCorporationInfo)
|
||||||
class EveCorporationInfoAdmin(admin.ModelAdmin):
|
class EveCorporationInfoAdmin(admin.ModelAdmin):
|
||||||
search_fields = ['corporation_name']
|
search_fields = ['corporation_name']
|
||||||
@@ -135,7 +183,7 @@ class EveCharacterAdmin(admin.ModelAdmin):
|
|||||||
'character_ownership__user__username'
|
'character_ownership__user__username'
|
||||||
]
|
]
|
||||||
list_display = (
|
list_display = (
|
||||||
'character_name', 'corporation_name', 'alliance_name', 'user', 'main_character'
|
'character_name', 'corporation_name', 'alliance_name', 'faction_name', 'user', 'main_character'
|
||||||
)
|
)
|
||||||
list_select_related = (
|
list_select_related = (
|
||||||
'character_ownership', 'character_ownership__user__profile__main_character'
|
'character_ownership', 'character_ownership__user__profile__main_character'
|
||||||
@@ -143,6 +191,7 @@ class EveCharacterAdmin(admin.ModelAdmin):
|
|||||||
list_filter = (
|
list_filter = (
|
||||||
'corporation_name',
|
'corporation_name',
|
||||||
'alliance_name',
|
'alliance_name',
|
||||||
|
'faction_name',
|
||||||
(
|
(
|
||||||
'character_ownership__user__profile__main_character',
|
'character_ownership__user__profile__main_character',
|
||||||
admin.RelatedOnlyFieldListFilter
|
admin.RelatedOnlyFieldListFilter
|
||||||
@@ -170,4 +219,4 @@ class EveCharacterAdmin(admin.ModelAdmin):
|
|||||||
def get_form(self, request, obj=None, **kwargs):
|
def get_form(self, request, obj=None, **kwargs):
|
||||||
if not obj or not obj.pk:
|
if not obj or not obj.pk:
|
||||||
return EveCharacterForm
|
return EveCharacterForm
|
||||||
return super(EveCharacterAdmin, self).get_form(request, obj=obj, **kwargs)
|
return super().get_form(request, obj=obj, **kwargs)
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ logger = logging.getLogger(__name__)
|
|||||||
|
|
||||||
def sync_user_groups(modeladmin, request, queryset):
|
def sync_user_groups(modeladmin, request, queryset):
|
||||||
for agc in queryset:
|
for agc in queryset:
|
||||||
logger.debug("update_all_states_group_membership for {}".format(agc))
|
logger.debug(f"update_all_states_group_membership for {agc}")
|
||||||
agc.update_all_states_group_membership()
|
agc.update_all_states_group_membership()
|
||||||
|
|
||||||
|
|
||||||
@@ -29,7 +29,7 @@ class AutogroupsConfigAdmin(admin.ModelAdmin):
|
|||||||
return []
|
return []
|
||||||
|
|
||||||
def get_actions(self, request):
|
def get_actions(self, request):
|
||||||
actions = super(AutogroupsConfigAdmin, self).get_actions(request)
|
actions = super().get_actions(request)
|
||||||
actions['sync_user_groups'] = (sync_user_groups,
|
actions['sync_user_groups'] = (sync_user_groups,
|
||||||
'sync_user_groups',
|
'sync_user_groups',
|
||||||
'Sync all users groups for this Autogroup Config')
|
'Sync all users groups for this Autogroup Config')
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.11.6 on 2017-12-23 04:30
|
# Generated by Django 1.11.6 on 2017-12-23 04:30
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
import django.db.models.deletion
|
import django.db.models.deletion
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ class AutogroupsConfigManager(models.Manager):
|
|||||||
for config in self.filter(states=state):
|
for config in self.filter(states=state):
|
||||||
logger.debug("in state loop")
|
logger.debug("in state loop")
|
||||||
for user in users:
|
for user in users:
|
||||||
logger.debug("in user loop for {}".format(user))
|
logger.debug(f"in user loop for {user}")
|
||||||
config.update_group_membership_for_user(user)
|
config.update_group_membership_for_user(user)
|
||||||
|
|
||||||
def update_groups_for_user(self, user: User, state: State = None):
|
def update_groups_for_user(self, user: User, state: State = None):
|
||||||
@@ -81,7 +81,7 @@ class AutogroupsConfig(models.Model):
|
|||||||
objects = AutogroupsConfigManager()
|
objects = AutogroupsConfigManager()
|
||||||
|
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
super(AutogroupsConfig, self).__init__(*args, **kwargs)
|
super().__init__(*args, **kwargs)
|
||||||
|
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return self.__class__.__name__
|
return self.__class__.__name__
|
||||||
@@ -111,24 +111,24 @@ class AutogroupsConfig(models.Model):
|
|||||||
group = None
|
group = None
|
||||||
try:
|
try:
|
||||||
if not self.alliance_groups or not self.user_entitled_to_groups(user):
|
if not self.alliance_groups or not self.user_entitled_to_groups(user):
|
||||||
logger.debug('User {} does not have required state for alliance group membership'.format(user))
|
logger.debug(f'User {user} does not have required state for alliance group membership')
|
||||||
return
|
return
|
||||||
else:
|
else:
|
||||||
alliance = user.profile.main_character.alliance
|
alliance = user.profile.main_character.alliance
|
||||||
if alliance is None:
|
if alliance is None:
|
||||||
logger.debug('User {} alliance is None, cannot update group membership'.format(user))
|
logger.debug(f'User {user} alliance is None, cannot update group membership')
|
||||||
return
|
return
|
||||||
group = self.get_alliance_group(alliance)
|
group = self.get_alliance_group(alliance)
|
||||||
except EveAllianceInfo.DoesNotExist:
|
except EveAllianceInfo.DoesNotExist:
|
||||||
logger.debug('User {} main characters alliance does not exist in the database. Creating.'.format(user))
|
logger.debug(f'User {user} main characters alliance does not exist in the database. Creating.')
|
||||||
alliance = EveAllianceInfo.objects.create_alliance(user.profile.main_character.alliance_id)
|
alliance = EveAllianceInfo.objects.create_alliance(user.profile.main_character.alliance_id)
|
||||||
group = self.get_alliance_group(alliance)
|
group = self.get_alliance_group(alliance)
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
logger.warning('User {} does not have a main character. Group membership not updated'.format(user))
|
logger.warning(f'User {user} does not have a main character. Group membership not updated')
|
||||||
finally:
|
finally:
|
||||||
self.remove_user_from_alliance_groups(user, except_group=group)
|
self.remove_user_from_alliance_groups(user, except_group=group)
|
||||||
if group is not None:
|
if group is not None:
|
||||||
logger.debug('Adding user {} to alliance group {}'.format(user, group))
|
logger.debug(f'Adding user {user} to alliance group {group}')
|
||||||
user.groups.add(group)
|
user.groups.add(group)
|
||||||
|
|
||||||
@transaction.atomic
|
@transaction.atomic
|
||||||
@@ -136,20 +136,20 @@ class AutogroupsConfig(models.Model):
|
|||||||
group = None
|
group = None
|
||||||
try:
|
try:
|
||||||
if not self.corp_groups or not self.user_entitled_to_groups(user):
|
if not self.corp_groups or not self.user_entitled_to_groups(user):
|
||||||
logger.debug('User {} does not have required state for corp group membership'.format(user))
|
logger.debug(f'User {user} does not have required state for corp group membership')
|
||||||
else:
|
else:
|
||||||
corp = user.profile.main_character.corporation
|
corp = user.profile.main_character.corporation
|
||||||
group = self.get_corp_group(corp)
|
group = self.get_corp_group(corp)
|
||||||
except EveCorporationInfo.DoesNotExist:
|
except EveCorporationInfo.DoesNotExist:
|
||||||
logger.debug('User {} main characters corporation does not exist in the database. Creating.'.format(user))
|
logger.debug(f'User {user} main characters corporation does not exist in the database. Creating.')
|
||||||
corp = EveCorporationInfo.objects.create_corporation(user.profile.main_character.corporation_id)
|
corp = EveCorporationInfo.objects.create_corporation(user.profile.main_character.corporation_id)
|
||||||
group = self.get_corp_group(corp)
|
group = self.get_corp_group(corp)
|
||||||
except AttributeError:
|
except AttributeError:
|
||||||
logger.warning('User {} does not have a main character. Group membership not updated'.format(user))
|
logger.warning(f'User {user} does not have a main character. Group membership not updated')
|
||||||
finally:
|
finally:
|
||||||
self.remove_user_from_corp_groups(user, except_group=group)
|
self.remove_user_from_corp_groups(user, except_group=group)
|
||||||
if group is not None:
|
if group is not None:
|
||||||
logger.debug('Adding user {} to corp group {}'.format(user, group))
|
logger.debug(f'Adding user {user} to corp group {group}')
|
||||||
user.groups.add(group)
|
user.groups.add(group)
|
||||||
|
|
||||||
@transaction.atomic
|
@transaction.atomic
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ def pre_save_config(sender, instance, *args, **kwargs):
|
|||||||
Checks if enable was toggled on group config and
|
Checks if enable was toggled on group config and
|
||||||
deletes groups if necessary.
|
deletes groups if necessary.
|
||||||
"""
|
"""
|
||||||
logger.debug("Received pre_save from {}".format(instance))
|
logger.debug(f"Received pre_save from {instance}")
|
||||||
if not instance.pk:
|
if not instance.pk:
|
||||||
# new model being created
|
# new model being created
|
||||||
return
|
return
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ MODULE_PATH = 'allianceauth.eveonline.autogroups'
|
|||||||
|
|
||||||
|
|
||||||
def patch(target, *args, **kwargs):
|
def patch(target, *args, **kwargs):
|
||||||
return mock.patch('{}{}'.format(MODULE_PATH, target), *args, **kwargs)
|
return mock.patch(f'{MODULE_PATH}{target}', *args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
def connect_signals():
|
def connect_signals():
|
||||||
|
|||||||
@@ -56,15 +56,15 @@ def _eve_entity_image_url(
|
|||||||
tenants = ['tranquility', 'singularity']
|
tenants = ['tranquility', 'singularity']
|
||||||
|
|
||||||
if not entity_id:
|
if not entity_id:
|
||||||
raise ValueError('Invalid entity_id: {}'.format(entity_id))
|
raise ValueError(f'Invalid entity_id: {entity_id}')
|
||||||
else:
|
else:
|
||||||
entity_id = int(entity_id)
|
entity_id = int(entity_id)
|
||||||
|
|
||||||
if not size or size < 32 or size > 1024 or (size & (size - 1) != 0):
|
if not size or size < 32 or size > 1024 or (size & (size - 1) != 0):
|
||||||
raise ValueError('Invalid size: {}'.format(size))
|
raise ValueError(f'Invalid size: {size}')
|
||||||
|
|
||||||
if category not in categories:
|
if category not in categories:
|
||||||
raise ValueError('Invalid category {}'.format(category))
|
raise ValueError(f'Invalid category {category}')
|
||||||
else:
|
else:
|
||||||
endpoint = categories[category]['endpoint']
|
endpoint = categories[category]['endpoint']
|
||||||
|
|
||||||
@@ -78,7 +78,7 @@ def _eve_entity_image_url(
|
|||||||
variant = categories[category]['variants'][0]
|
variant = categories[category]['variants'][0]
|
||||||
|
|
||||||
if tenant and tenant not in tenants:
|
if tenant and tenant not in tenants:
|
||||||
raise ValueError('Invalid tenant {}'.format(tenant))
|
raise ValueError(f'Invalid tenant {tenant}')
|
||||||
|
|
||||||
# compose result URL
|
# compose result URL
|
||||||
result = '{}/{}/{}/{}?size={}'.format(
|
result = '{}/{}/{}/{}?size={}'.format(
|
||||||
@@ -89,7 +89,7 @@ def _eve_entity_image_url(
|
|||||||
size
|
size
|
||||||
)
|
)
|
||||||
if tenant:
|
if tenant:
|
||||||
result += '&tenant={}'.format(tenant)
|
result += f'&tenant={tenant}'
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ def _build_url(category: str, eve_id: int) -> str:
|
|||||||
|
|
||||||
url = urljoin(
|
url = urljoin(
|
||||||
_BASE_URL,
|
_BASE_URL,
|
||||||
'{}/{}'.format(partial, int(eve_id))
|
f'{partial}/{int(eve_id)}'
|
||||||
)
|
)
|
||||||
return url
|
return url
|
||||||
|
|
||||||
|
|||||||
@@ -39,7 +39,7 @@ def _build_url(category: str, eve_id: int) -> str:
|
|||||||
|
|
||||||
url = urljoin(
|
url = urljoin(
|
||||||
_BASE_URL,
|
_BASE_URL,
|
||||||
'{}/{}/'.format(partial, int(eve_id))
|
f'{partial}/{int(eve_id)}/'
|
||||||
)
|
)
|
||||||
return url
|
return url
|
||||||
|
|
||||||
|
|||||||
@@ -27,6 +27,8 @@ class EveCharacterManager(models.Manager):
|
|||||||
alliance_id=character.alliance.id,
|
alliance_id=character.alliance.id,
|
||||||
alliance_name=character.alliance.name,
|
alliance_name=character.alliance.name,
|
||||||
alliance_ticker=getattr(character.alliance, 'ticker', None),
|
alliance_ticker=getattr(character.alliance, 'ticker', None),
|
||||||
|
faction_id=character.faction.id,
|
||||||
|
faction_name=character.faction.name
|
||||||
)
|
)
|
||||||
|
|
||||||
def update_character(self, character_id):
|
def update_character(self, character_id):
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-05 21:39
|
# Generated by Django 1.10.1 on 2016-09-05 21:39
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-09-10 20:20
|
# Generated by Django 1.10.1 on 2016-09-10 20:20
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.2 on 2016-10-26 01:49
|
# Generated by Django 1.10.2 on 2016-10-26 01:49
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.conf import settings
|
from django.conf import settings
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.2 on 2016-11-01 04:20
|
# Generated by Django 1.10.2 on 2016-11-01 04:20
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2016-12-16 23:22
|
# Generated by Django 1.10.1 on 2016-12-16 23:22
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.1 on 2017-01-02 19:23
|
# Generated by Django 1.10.1 on 2017-01-02 19:23
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|||||||
@@ -1,11 +1,10 @@
|
|||||||
# Generated by Django 1.10.5 on 2017-01-18 13:20
|
# Generated by Django 1.10.5 on 2017-01-18 13:20
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
def get_duplicates(items):
|
def get_duplicates(items):
|
||||||
return set([item for item in items if items.count(item) > 1])
|
return {item for item in items if items.count(item) > 1}
|
||||||
|
|
||||||
|
|
||||||
def enforce_unique_characters(apps, schema_editor):
|
def enforce_unique_characters(apps, schema_editor):
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.10.5 on 2017-03-22 23:09
|
# Generated by Django 1.10.5 on 2017-03-22 23:09
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations
|
from django.db import migrations
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
# Generated by Django 1.11.5 on 2017-09-28 02:16
|
# Generated by Django 1.11.5 on 2017-09-28 02:16
|
||||||
from __future__ import unicode_literals
|
|
||||||
|
|
||||||
from django.db import migrations, models
|
from django.db import migrations, models
|
||||||
import django.db.models.deletion
|
import django.db.models.deletion
|
||||||
|
|||||||
35
allianceauth/eveonline/migrations/0015_factions.py
Normal file
35
allianceauth/eveonline/migrations/0015_factions.py
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
# Generated by Django 3.1.13 on 2021-10-12 01:04
|
||||||
|
|
||||||
|
from django.db import migrations, models
|
||||||
|
|
||||||
|
|
||||||
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
|
dependencies = [
|
||||||
|
('eveonline', '0014_auto_20210105_1413'),
|
||||||
|
]
|
||||||
|
|
||||||
|
operations = [
|
||||||
|
migrations.CreateModel(
|
||||||
|
name='EveFactionInfo',
|
||||||
|
fields=[
|
||||||
|
('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||||||
|
('faction_id', models.PositiveIntegerField(db_index=True, unique=True)),
|
||||||
|
('faction_name', models.CharField(max_length=254, unique=True)),
|
||||||
|
],
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='evecharacter',
|
||||||
|
name='faction_id',
|
||||||
|
field=models.PositiveIntegerField(blank=True, default=None, null=True),
|
||||||
|
),
|
||||||
|
migrations.AddField(
|
||||||
|
model_name='evecharacter',
|
||||||
|
name='faction_name',
|
||||||
|
field=models.CharField(blank=True, default='', max_length=254, null=True),
|
||||||
|
),
|
||||||
|
migrations.AddIndex(
|
||||||
|
model_name='evecharacter',
|
||||||
|
index=models.Index(fields=['faction_id'], name='eveonline_e_faction_d5274e_idx'),
|
||||||
|
),
|
||||||
|
]
|
||||||
@@ -1,13 +1,68 @@
|
|||||||
from django.db import models
|
import logging
|
||||||
from typing import Union
|
from typing import Union
|
||||||
|
|
||||||
from .managers import EveCharacterManager, EveCharacterProviderManager
|
from django.core.exceptions import ObjectDoesNotExist
|
||||||
from .managers import EveCorporationManager, EveCorporationProviderManager
|
from django.db import models
|
||||||
from .managers import EveAllianceManager, EveAllianceProviderManager
|
from esi.models import Token
|
||||||
|
|
||||||
|
from allianceauth.notifications import notify
|
||||||
|
|
||||||
from . import providers
|
from . import providers
|
||||||
from .evelinks import eveimageserver
|
from .evelinks import eveimageserver
|
||||||
|
from .managers import (
|
||||||
|
EveAllianceManager,
|
||||||
|
EveAllianceProviderManager,
|
||||||
|
EveCharacterManager,
|
||||||
|
EveCharacterProviderManager,
|
||||||
|
EveCorporationManager,
|
||||||
|
EveCorporationProviderManager,
|
||||||
|
)
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
_DEFAULT_IMAGE_SIZE = 32
|
_DEFAULT_IMAGE_SIZE = 32
|
||||||
|
DOOMHEIM_CORPORATION_ID = 1000001
|
||||||
|
|
||||||
|
|
||||||
|
class EveFactionInfo(models.Model):
|
||||||
|
faction_id = models.PositiveIntegerField(unique=True, db_index=True)
|
||||||
|
faction_name = models.CharField(max_length=254, unique=True)
|
||||||
|
|
||||||
|
provider = providers.provider
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return self.faction_name
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def generic_logo_url(
|
||||||
|
faction_id: int, size: int = _DEFAULT_IMAGE_SIZE
|
||||||
|
) -> str:
|
||||||
|
"""image URL for the given faction ID"""
|
||||||
|
return eveimageserver.corporation_logo_url(faction_id, size)
|
||||||
|
|
||||||
|
def logo_url(self, size: int = _DEFAULT_IMAGE_SIZE) -> str:
|
||||||
|
"""image URL of this faction"""
|
||||||
|
return self.generic_logo_url(self.faction_id, size)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def logo_url_32(self) -> str:
|
||||||
|
"""image URL for this faction"""
|
||||||
|
return self.logo_url(32)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def logo_url_64(self) -> str:
|
||||||
|
"""image URL for this faction"""
|
||||||
|
return self.logo_url(64)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def logo_url_128(self) -> str:
|
||||||
|
"""image URL for this faction"""
|
||||||
|
return self.logo_url(128)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def logo_url_256(self) -> str:
|
||||||
|
"""image URL for this faction"""
|
||||||
|
return self.logo_url(256)
|
||||||
|
|
||||||
|
|
||||||
class EveAllianceInfo(models.Model):
|
class EveAllianceInfo(models.Model):
|
||||||
@@ -27,13 +82,12 @@ class EveAllianceInfo(models.Model):
|
|||||||
for corp_id in alliance.corp_ids:
|
for corp_id in alliance.corp_ids:
|
||||||
if not EveCorporationInfo.objects.filter(corporation_id=corp_id).exists():
|
if not EveCorporationInfo.objects.filter(corporation_id=corp_id).exists():
|
||||||
EveCorporationInfo.objects.create_corporation(corp_id)
|
EveCorporationInfo.objects.create_corporation(corp_id)
|
||||||
EveCorporationInfo.objects.filter(
|
EveCorporationInfo.objects.filter(corporation_id__in=alliance.corp_ids).update(
|
||||||
corporation_id__in=alliance.corp_ids).update(alliance=self
|
alliance=self
|
||||||
)
|
)
|
||||||
EveCorporationInfo.objects\
|
EveCorporationInfo.objects.filter(alliance=self).exclude(
|
||||||
.filter(alliance=self)\
|
corporation_id__in=alliance.corp_ids
|
||||||
.exclude(corporation_id__in=alliance.corp_ids)\
|
).update(alliance=None)
|
||||||
.update(alliance=None)
|
|
||||||
|
|
||||||
def update_alliance(self, alliance: providers.Alliance = None):
|
def update_alliance(self, alliance: providers.Alliance = None):
|
||||||
if alliance is None:
|
if alliance is None:
|
||||||
@@ -141,6 +195,7 @@ class EveCorporationInfo(models.Model):
|
|||||||
|
|
||||||
|
|
||||||
class EveCharacter(models.Model):
|
class EveCharacter(models.Model):
|
||||||
|
"""Character in Eve Online"""
|
||||||
character_id = models.PositiveIntegerField(unique=True)
|
character_id = models.PositiveIntegerField(unique=True)
|
||||||
character_name = models.CharField(max_length=254, unique=True)
|
character_name = models.CharField(max_length=254, unique=True)
|
||||||
corporation_id = models.PositiveIntegerField()
|
corporation_id = models.PositiveIntegerField()
|
||||||
@@ -149,6 +204,8 @@ class EveCharacter(models.Model):
|
|||||||
alliance_id = models.PositiveIntegerField(blank=True, null=True, default=None)
|
alliance_id = models.PositiveIntegerField(blank=True, null=True, default=None)
|
||||||
alliance_name = models.CharField(max_length=254, blank=True, null=True, default='')
|
alliance_name = models.CharField(max_length=254, blank=True, null=True, default='')
|
||||||
alliance_ticker = models.CharField(max_length=5, blank=True, null=True, default='')
|
alliance_ticker = models.CharField(max_length=5, blank=True, null=True, default='')
|
||||||
|
faction_id = models.PositiveIntegerField(blank=True, null=True, default=None)
|
||||||
|
faction_name = models.CharField(max_length=254, blank=True, null=True, default='')
|
||||||
|
|
||||||
objects = EveCharacterManager()
|
objects = EveCharacterManager()
|
||||||
provider = EveCharacterProviderManager()
|
provider = EveCharacterProviderManager()
|
||||||
@@ -159,8 +216,17 @@ class EveCharacter(models.Model):
|
|||||||
models.Index(fields=['alliance_id',]),
|
models.Index(fields=['alliance_id',]),
|
||||||
models.Index(fields=['corporation_name',]),
|
models.Index(fields=['corporation_name',]),
|
||||||
models.Index(fields=['alliance_name',]),
|
models.Index(fields=['alliance_name',]),
|
||||||
|
models.Index(fields=['faction_id',]),
|
||||||
]
|
]
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return self.character_name
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_biomassed(self) -> bool:
|
||||||
|
"""Whether this character is dead or not."""
|
||||||
|
return self.corporation_id == DOOMHEIM_CORPORATION_ID
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def alliance(self) -> Union[EveAllianceInfo, None]:
|
def alliance(self) -> Union[EveAllianceInfo, None]:
|
||||||
"""
|
"""
|
||||||
@@ -181,6 +247,17 @@ class EveCharacter(models.Model):
|
|||||||
"""
|
"""
|
||||||
return EveCorporationInfo.objects.get(corporation_id=self.corporation_id)
|
return EveCorporationInfo.objects.get(corporation_id=self.corporation_id)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def faction(self) -> Union[EveFactionInfo, None]:
|
||||||
|
"""
|
||||||
|
Pseudo foreign key from faction_id to EveFactionInfo
|
||||||
|
:raises: EveFactionInfo.DoesNotExist
|
||||||
|
:return: EveFactionInfo
|
||||||
|
"""
|
||||||
|
if self.faction_id is None:
|
||||||
|
return None
|
||||||
|
return EveFactionInfo.objects.get(faction_id=self.faction_id)
|
||||||
|
|
||||||
def update_character(self, character: providers.Character = None):
|
def update_character(self, character: providers.Character = None):
|
||||||
if character is None:
|
if character is None:
|
||||||
character = self.provider.get_character(self.character_id)
|
character = self.provider.get_character(self.character_id)
|
||||||
@@ -191,11 +268,39 @@ class EveCharacter(models.Model):
|
|||||||
self.alliance_id = character.alliance.id
|
self.alliance_id = character.alliance.id
|
||||||
self.alliance_name = character.alliance.name
|
self.alliance_name = character.alliance.name
|
||||||
self.alliance_ticker = getattr(character.alliance, 'ticker', None)
|
self.alliance_ticker = getattr(character.alliance, 'ticker', None)
|
||||||
|
self.faction_id = character.faction.id
|
||||||
|
self.faction_name = character.faction.name
|
||||||
self.save()
|
self.save()
|
||||||
|
if self.is_biomassed:
|
||||||
|
self._remove_tokens_of_biomassed_character()
|
||||||
return self
|
return self
|
||||||
|
|
||||||
def __str__(self):
|
def _remove_tokens_of_biomassed_character(self) -> None:
|
||||||
return self.character_name
|
"""Remove tokens of this biomassed character."""
|
||||||
|
try:
|
||||||
|
user = self.character_ownership.user
|
||||||
|
except ObjectDoesNotExist:
|
||||||
|
return
|
||||||
|
tokens_to_delete = Token.objects.filter(character_id=self.character_id)
|
||||||
|
tokens_count = tokens_to_delete.count()
|
||||||
|
if not tokens_count:
|
||||||
|
return
|
||||||
|
tokens_to_delete.delete()
|
||||||
|
logger.info(
|
||||||
|
"%d tokens from user %s for biomassed character %s [id:%s] deleted.",
|
||||||
|
tokens_count,
|
||||||
|
user,
|
||||||
|
self,
|
||||||
|
self.character_id,
|
||||||
|
)
|
||||||
|
notify(
|
||||||
|
user=user,
|
||||||
|
title=f"Character {self} biomassed",
|
||||||
|
message=(
|
||||||
|
f"Your former character {self} has been biomassed "
|
||||||
|
"and has been removed from the list of your alts."
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def generic_portrait_url(
|
def generic_portrait_url(
|
||||||
@@ -278,3 +383,30 @@ class EveCharacter(models.Model):
|
|||||||
def alliance_logo_url_256(self) -> str:
|
def alliance_logo_url_256(self) -> str:
|
||||||
"""image URL for alliance of this character or empty string"""
|
"""image URL for alliance of this character or empty string"""
|
||||||
return self.alliance_logo_url(256)
|
return self.alliance_logo_url(256)
|
||||||
|
|
||||||
|
def faction_logo_url(self, size=_DEFAULT_IMAGE_SIZE) -> str:
|
||||||
|
"""image URL for alliance of this character or empty string"""
|
||||||
|
if self.faction_id:
|
||||||
|
return EveFactionInfo.generic_logo_url(self.faction_id, size)
|
||||||
|
else:
|
||||||
|
return ''
|
||||||
|
|
||||||
|
@property
|
||||||
|
def faction_logo_url_32(self) -> str:
|
||||||
|
"""image URL for alliance of this character or empty string"""
|
||||||
|
return self.faction_logo_url(32)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def faction_logo_url_64(self) -> str:
|
||||||
|
"""image URL for alliance of this character or empty string"""
|
||||||
|
return self.faction_logo_url(64)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def faction_logo_url_128(self) -> str:
|
||||||
|
"""image URL for alliance of this character or empty string"""
|
||||||
|
return self.faction_logo_url(128)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def faction_logo_url_256(self) -> str:
|
||||||
|
"""image URL for alliance of this character or empty string"""
|
||||||
|
return self.faction_logo_url(256)
|
||||||
|
|||||||
@@ -13,16 +13,18 @@ from allianceauth import __version__
|
|||||||
SWAGGER_SPEC_PATH = os.path.join(os.path.dirname(
|
SWAGGER_SPEC_PATH = os.path.join(os.path.dirname(
|
||||||
os.path.abspath(__file__)), 'swagger.json'
|
os.path.abspath(__file__)), 'swagger.json'
|
||||||
)
|
)
|
||||||
"""
|
|
||||||
Swagger spec operations:
|
|
||||||
|
|
||||||
get_alliances_alliance_id
|
# for the love of Bob please add operations you use here. I'm tired of breaking undocumented things.
|
||||||
get_alliances_alliance_id_corporations
|
ESI_OPERATIONS=[
|
||||||
get_corporations_corporation_id
|
'get_alliances_alliance_id',
|
||||||
get_characters_character_id
|
'get_alliances_alliance_id_corporations',
|
||||||
get_universe_types_type_id
|
'get_corporations_corporation_id',
|
||||||
post_character_affiliation
|
'get_characters_character_id',
|
||||||
"""
|
'post_characters_affiliation',
|
||||||
|
'get_universe_types_type_id',
|
||||||
|
'get_universe_factions',
|
||||||
|
'post_universe_names',
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
@@ -34,11 +36,12 @@ class ObjectNotFound(Exception):
|
|||||||
self.type = type_name
|
self.type = type_name
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return '%s with ID %s not found.' % (self.type, self.id)
|
return f'{self.type} with ID {self.id} not found.'
|
||||||
|
|
||||||
|
|
||||||
class Entity(object):
|
class Entity:
|
||||||
def __init__(self, id=None, name=None):
|
def __init__(self, id=None, name=None, **kwargs):
|
||||||
|
super().__init__(**kwargs)
|
||||||
self.id = id
|
self.id = id
|
||||||
self.name = name
|
self.name = name
|
||||||
|
|
||||||
@@ -46,7 +49,7 @@ class Entity(object):
|
|||||||
return self.name
|
return self.name
|
||||||
|
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return "<{} ({}): {}>".format(self.__class__.__name__, self.id, self.name)
|
return f"<{self.__class__.__name__} ({self.id}): {self.name}>"
|
||||||
|
|
||||||
def __bool__(self):
|
def __bool__(self):
|
||||||
return bool(self.id)
|
return bool(self.id)
|
||||||
@@ -55,15 +58,11 @@ class Entity(object):
|
|||||||
return self.id == other.id
|
return self.id == other.id
|
||||||
|
|
||||||
|
|
||||||
class Corporation(Entity):
|
class AllianceMixin:
|
||||||
def __init__(self, ticker=None, ceo_id=None, members=None, alliance_id=None, **kwargs):
|
def __init__(self, alliance_id=None, **kwargs):
|
||||||
super(Corporation, self).__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
self.ticker = ticker
|
|
||||||
self.ceo_id = ceo_id
|
|
||||||
self.members = members
|
|
||||||
self.alliance_id = alliance_id
|
self.alliance_id = alliance_id
|
||||||
self._alliance = None
|
self._alliance = None
|
||||||
self._ceo = None
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def alliance(self):
|
def alliance(self):
|
||||||
@@ -73,6 +72,30 @@ class Corporation(Entity):
|
|||||||
return self._alliance
|
return self._alliance
|
||||||
return Entity(None, None)
|
return Entity(None, None)
|
||||||
|
|
||||||
|
|
||||||
|
class FactionMixin:
|
||||||
|
def __init__(self, faction_id=None, **kwargs):
|
||||||
|
super().__init__(**kwargs)
|
||||||
|
self.faction_id = faction_id
|
||||||
|
self._faction = None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def faction(self):
|
||||||
|
if self.faction_id:
|
||||||
|
if not self._faction:
|
||||||
|
self._faction = provider.get_faction(self.faction_id)
|
||||||
|
return self._faction
|
||||||
|
return Entity(None, None)
|
||||||
|
|
||||||
|
|
||||||
|
class Corporation(Entity, AllianceMixin, FactionMixin):
|
||||||
|
def __init__(self, ticker=None, ceo_id=None, members=None, **kwargs):
|
||||||
|
super().__init__(**kwargs)
|
||||||
|
self.ticker = ticker
|
||||||
|
self.ceo_id = ceo_id
|
||||||
|
self.members = members
|
||||||
|
self._ceo = None
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def ceo(self):
|
def ceo(self):
|
||||||
if not self._ceo:
|
if not self._ceo:
|
||||||
@@ -80,9 +103,9 @@ class Corporation(Entity):
|
|||||||
return self._ceo
|
return self._ceo
|
||||||
|
|
||||||
|
|
||||||
class Alliance(Entity):
|
class Alliance(Entity, FactionMixin):
|
||||||
def __init__(self, ticker=None, corp_ids=None, executor_corp_id=None, **kwargs):
|
def __init__(self, ticker=None, corp_ids=None, executor_corp_id=None, **kwargs):
|
||||||
super(Alliance, self).__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
self.ticker = ticker
|
self.ticker = ticker
|
||||||
self.corp_ids = corp_ids
|
self.corp_ids = corp_ids
|
||||||
self.executor_corp_id = executor_corp_id
|
self.executor_corp_id = executor_corp_id
|
||||||
@@ -97,7 +120,7 @@ class Alliance(Entity):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def corps(self):
|
def corps(self):
|
||||||
return sorted([self.corp(c_id) for c_id in self.corp_ids], key=lambda x: x.name)
|
return sorted((self.corp(c_id) for c_id in self.corp_ids), key=lambda x: x.name)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def executor_corp(self):
|
def executor_corp(self):
|
||||||
@@ -106,13 +129,11 @@ class Alliance(Entity):
|
|||||||
return Entity(None, None)
|
return Entity(None, None)
|
||||||
|
|
||||||
|
|
||||||
class Character(Entity):
|
class Character(Entity, AllianceMixin, FactionMixin):
|
||||||
def __init__(self, corp_id=None, alliance_id=None, **kwargs):
|
def __init__(self, corp_id=None, **kwargs):
|
||||||
super(Character, self).__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
self.corp_id = corp_id
|
self.corp_id = corp_id
|
||||||
self.alliance_id = alliance_id
|
|
||||||
self._corp = None
|
self._corp = None
|
||||||
self._alliance = None
|
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def corp(self):
|
def corp(self):
|
||||||
@@ -120,19 +141,13 @@ class Character(Entity):
|
|||||||
self._corp = provider.get_corp(self.corp_id)
|
self._corp = provider.get_corp(self.corp_id)
|
||||||
return self._corp
|
return self._corp
|
||||||
|
|
||||||
@property
|
|
||||||
def alliance(self):
|
|
||||||
if self.alliance_id:
|
|
||||||
return self.corp.alliance
|
|
||||||
return Entity(None, None)
|
|
||||||
|
|
||||||
|
|
||||||
class ItemType(Entity):
|
class ItemType(Entity):
|
||||||
def __init__(self, **kwargs):
|
def __init__(self, **kwargs):
|
||||||
super(ItemType, self).__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
|
|
||||||
|
|
||||||
class EveProvider(object):
|
class EveProvider:
|
||||||
def get_alliance(self, alliance_id):
|
def get_alliance(self, alliance_id):
|
||||||
"""
|
"""
|
||||||
:return: an Alliance object for the given ID
|
:return: an Alliance object for the given ID
|
||||||
@@ -155,7 +170,7 @@ class EveProvider(object):
|
|||||||
"""
|
"""
|
||||||
:return: an ItemType object for the given ID
|
:return: an ItemType object for the given ID
|
||||||
"""
|
"""
|
||||||
raise NotImplemented()
|
raise NotImplementedError()
|
||||||
|
|
||||||
|
|
||||||
class EveSwaggerProvider(EveProvider):
|
class EveSwaggerProvider(EveProvider):
|
||||||
@@ -179,6 +194,7 @@ class EveSwaggerProvider(EveProvider):
|
|||||||
|
|
||||||
self._token = token
|
self._token = token
|
||||||
self.adapter = adapter or self
|
self.adapter = adapter or self
|
||||||
|
self._faction_list = None # what are the odds this will change? could cache forever!
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def client(self):
|
def client(self):
|
||||||
@@ -191,7 +207,8 @@ class EveSwaggerProvider(EveProvider):
|
|||||||
def __str__(self):
|
def __str__(self):
|
||||||
return 'esi'
|
return 'esi'
|
||||||
|
|
||||||
def get_alliance(self, alliance_id):
|
def get_alliance(self, alliance_id: int) -> Alliance:
|
||||||
|
"""Fetch alliance from ESI."""
|
||||||
try:
|
try:
|
||||||
data = self.client.Alliance.get_alliances_alliance_id(alliance_id=alliance_id).result()
|
data = self.client.Alliance.get_alliances_alliance_id(alliance_id=alliance_id).result()
|
||||||
corps = self.client.Alliance.get_alliances_alliance_id_corporations(alliance_id=alliance_id).result()
|
corps = self.client.Alliance.get_alliances_alliance_id_corporations(alliance_id=alliance_id).result()
|
||||||
@@ -201,12 +218,14 @@ class EveSwaggerProvider(EveProvider):
|
|||||||
ticker=data['ticker'],
|
ticker=data['ticker'],
|
||||||
corp_ids=corps,
|
corp_ids=corps,
|
||||||
executor_corp_id=data['executor_corporation_id'] if 'executor_corporation_id' in data else None,
|
executor_corp_id=data['executor_corporation_id'] if 'executor_corporation_id' in data else None,
|
||||||
|
faction_id=data['faction_id'] if 'faction_id' in data else None,
|
||||||
)
|
)
|
||||||
return model
|
return model
|
||||||
except HTTPNotFound:
|
except HTTPNotFound:
|
||||||
raise ObjectNotFound(alliance_id, 'alliance')
|
raise ObjectNotFound(alliance_id, 'alliance')
|
||||||
|
|
||||||
def get_corp(self, corp_id):
|
def get_corp(self, corp_id: int) -> Corporation:
|
||||||
|
"""Fetch corporation from ESI."""
|
||||||
try:
|
try:
|
||||||
data = self.client.Corporation.get_corporations_corporation_id(corporation_id=corp_id).result()
|
data = self.client.Corporation.get_corporations_corporation_id(corporation_id=corp_id).result()
|
||||||
model = Corporation(
|
model = Corporation(
|
||||||
@@ -216,27 +235,62 @@ class EveSwaggerProvider(EveProvider):
|
|||||||
ceo_id=data['ceo_id'],
|
ceo_id=data['ceo_id'],
|
||||||
members=data['member_count'],
|
members=data['member_count'],
|
||||||
alliance_id=data['alliance_id'] if 'alliance_id' in data else None,
|
alliance_id=data['alliance_id'] if 'alliance_id' in data else None,
|
||||||
|
faction_id=data['faction_id'] if 'faction_id' in data else None,
|
||||||
)
|
)
|
||||||
return model
|
return model
|
||||||
except HTTPNotFound:
|
except HTTPNotFound:
|
||||||
raise ObjectNotFound(corp_id, 'corporation')
|
raise ObjectNotFound(corp_id, 'corporation')
|
||||||
|
|
||||||
def get_character(self, character_id):
|
def get_character(self, character_id: int) -> Character:
|
||||||
|
"""Fetch character from ESI."""
|
||||||
try:
|
try:
|
||||||
data = self.client.Character.get_characters_character_id(character_id=character_id).result()
|
character_name = self._fetch_character_name(character_id)
|
||||||
affiliation = self.client.Character.post_characters_affiliation(characters=[character_id]).result()[0]
|
affiliation = self.client.Character.post_characters_affiliation(characters=[character_id]).result()[0]
|
||||||
|
|
||||||
model = Character(
|
model = Character(
|
||||||
id=character_id,
|
id=character_id,
|
||||||
name=data['name'],
|
name=character_name,
|
||||||
corp_id=affiliation['corporation_id'],
|
corp_id=affiliation['corporation_id'],
|
||||||
alliance_id=affiliation['alliance_id'] if 'alliance_id' in affiliation else None,
|
alliance_id=affiliation['alliance_id'] if 'alliance_id' in affiliation else None,
|
||||||
|
faction_id=affiliation['faction_id'] if 'faction_id' in affiliation else None,
|
||||||
)
|
)
|
||||||
return model
|
return model
|
||||||
except (HTTPNotFound, HTTPUnprocessableEntity):
|
except (HTTPNotFound, HTTPUnprocessableEntity, ObjectNotFound):
|
||||||
raise ObjectNotFound(character_id, 'character')
|
raise ObjectNotFound(character_id, 'character')
|
||||||
|
|
||||||
def get_itemtype(self, type_id):
|
def _fetch_character_name(self, character_id: int) -> str:
|
||||||
|
"""Fetch character name from ESI."""
|
||||||
|
data = self.client.Universe.post_universe_names(ids=[character_id]).result()
|
||||||
|
character = data.pop() if data else None
|
||||||
|
if (
|
||||||
|
not character
|
||||||
|
or character["category"] != "character"
|
||||||
|
or character["id"] != character_id
|
||||||
|
):
|
||||||
|
raise ObjectNotFound(character_id, 'character')
|
||||||
|
return character["name"]
|
||||||
|
|
||||||
|
def get_all_factions(self):
|
||||||
|
"""Fetch all factions from ESI."""
|
||||||
|
if not self._faction_list:
|
||||||
|
self._faction_list = self.client.Universe.get_universe_factions().result()
|
||||||
|
return self._faction_list
|
||||||
|
|
||||||
|
def get_faction(self, faction_id: int):
|
||||||
|
"""Fetch faction from ESI."""
|
||||||
|
faction_id = int(faction_id)
|
||||||
|
try:
|
||||||
|
if not self._faction_list:
|
||||||
|
_ = self.get_all_factions()
|
||||||
|
for f in self._faction_list:
|
||||||
|
if f['faction_id'] == faction_id:
|
||||||
|
return Entity(id=f['faction_id'], name=f['name'])
|
||||||
|
else:
|
||||||
|
raise KeyError()
|
||||||
|
except (HTTPNotFound, HTTPUnprocessableEntity, KeyError):
|
||||||
|
raise ObjectNotFound(faction_id, 'faction')
|
||||||
|
|
||||||
|
def get_itemtype(self, type_id: int) -> ItemType:
|
||||||
|
"""Fetch inventory item from ESI."""
|
||||||
try:
|
try:
|
||||||
data = self.client.Universe.get_universe_types_type_id(type_id=type_id).result()
|
data = self.client.Universe.get_universe_types_type_id(type_id=type_id).result()
|
||||||
return ItemType(id=type_id, name=data['name'])
|
return ItemType(id=type_id, name=data['name'])
|
||||||
|
|||||||
File diff suppressed because one or more lines are too long
@@ -1,12 +1,11 @@
|
|||||||
import logging
|
import logging
|
||||||
|
|
||||||
from celery import shared_task
|
from celery import shared_task
|
||||||
from .models import EveAllianceInfo
|
|
||||||
from .models import EveCharacter
|
|
||||||
from .models import EveCorporationInfo
|
|
||||||
|
|
||||||
|
from .models import EveAllianceInfo, EveCharacter, EveCorporationInfo
|
||||||
from . import providers
|
from . import providers
|
||||||
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
TASK_PRIORITY = 7
|
TASK_PRIORITY = 7
|
||||||
@@ -32,8 +31,8 @@ def update_alliance(alliance_id):
|
|||||||
|
|
||||||
|
|
||||||
@shared_task
|
@shared_task
|
||||||
def update_character(character_id):
|
def update_character(character_id: int) -> None:
|
||||||
"""Update given character from ESI"""
|
"""Update given character from ESI."""
|
||||||
EveCharacter.objects.update_character(character_id)
|
EveCharacter.objects.update_character(character_id)
|
||||||
|
|
||||||
|
|
||||||
@@ -43,23 +42,35 @@ def run_model_update():
|
|||||||
|
|
||||||
# update existing corp models
|
# update existing corp models
|
||||||
for corp in EveCorporationInfo.objects.all().values('corporation_id'):
|
for corp in EveCorporationInfo.objects.all().values('corporation_id'):
|
||||||
update_corp.apply_async(
|
update_corp.apply_async(args=[corp['corporation_id']], priority=TASK_PRIORITY)
|
||||||
args=[corp['corporation_id']], priority=TASK_PRIORITY
|
|
||||||
)
|
|
||||||
|
|
||||||
# update existing alliance models
|
# update existing alliance models
|
||||||
for alliance in EveAllianceInfo.objects.all().values('alliance_id'):
|
for alliance in EveAllianceInfo.objects.all().values('alliance_id'):
|
||||||
update_alliance.apply_async(
|
update_alliance.apply_async(args=[alliance['alliance_id']], priority=TASK_PRIORITY)
|
||||||
args=[alliance['alliance_id']], priority=TASK_PRIORITY
|
|
||||||
)
|
|
||||||
|
|
||||||
# update existing character models
|
# update existing character models
|
||||||
character_ids = EveCharacter.objects.all().values_list('character_id', flat=True)
|
character_ids = EveCharacter.objects.all().values_list('character_id', flat=True)
|
||||||
for character_ids_chunk in chunks(character_ids, CHUNK_SIZE):
|
for character_ids_chunk in chunks(character_ids, CHUNK_SIZE):
|
||||||
|
update_character_chunk.apply_async(
|
||||||
|
args=[character_ids_chunk], priority=TASK_PRIORITY
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@shared_task
|
||||||
|
def update_character_chunk(character_ids_chunk: list):
|
||||||
|
"""Update a list of character from ESI"""
|
||||||
|
try:
|
||||||
affiliations_raw = providers.provider.client.Character\
|
affiliations_raw = providers.provider.client.Character\
|
||||||
.post_characters_affiliation(characters=character_ids_chunk).result()
|
.post_characters_affiliation(characters=character_ids_chunk).result()
|
||||||
character_names = providers.provider.client.Universe\
|
character_names = providers.provider.client.Universe\
|
||||||
.post_universe_names(ids=character_ids_chunk).result()
|
.post_universe_names(ids=character_ids_chunk).result()
|
||||||
|
except OSError:
|
||||||
|
logger.info("Failed to bulk update characters. Attempting single updates")
|
||||||
|
for character_id in character_ids_chunk:
|
||||||
|
update_character.apply_async(
|
||||||
|
args=[character_id], priority=TASK_PRIORITY
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
affiliations = {
|
affiliations = {
|
||||||
affiliation.get('character_id'): affiliation
|
affiliation.get('character_id'): affiliation
|
||||||
|
|||||||
@@ -18,7 +18,7 @@ def set_logger(logger_name: str, name: str) -> object:
|
|||||||
'%(asctime)s - %(levelname)s - %(module)s:%(funcName)s - %(message)s'
|
'%(asctime)s - %(levelname)s - %(module)s:%(funcName)s - %(message)s'
|
||||||
)
|
)
|
||||||
f_handler = logging.FileHandler(
|
f_handler = logging.FileHandler(
|
||||||
'{}.log'.format(os.path.splitext(name)[0]),
|
f'{os.path.splitext(name)[0]}.log',
|
||||||
'w+'
|
'w+'
|
||||||
)
|
)
|
||||||
f_handler.setFormatter(f_format)
|
f_handler.setFormatter(f_format)
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user