dribdat/dribdat
1
0
Fork 0
mirror of https://codeberg.org/dribdat/dribdat.git synced 2026-07-13 21:39:15 +00:00
Code Issues 3 Projects Releases Packages Wiki Activity

cli db safety

Browse source
This commit is contained in:
Oleg Lavrovsky 2026-06-20 23:02:49 +02:00
commit 67f32fb794
No known key found for this signature in database
GPG key ID: 31E523030632FF4B
1 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

18
cli.py
View file

@ -10,6 +10,7 @@ from dribdat.user.models import User, Event
from dribdat.settings import DevConfig, ProdConfig
from dribdat.apipackage import fetch_datapackage, load_file_datapackage, import_users_csv
from dribdat.futures import UTC
from psycopg2.errors import ForeignKeyViolation
HERE = os.path.abspath(os.path.dirname(__file__))
TEST_PATH = os.path.join(HERE, 'tests')
@ -224,19 +225,26 @@ def kick(lowscore: bool, inactive: bool, withsso: bool, delete: bool, score: int
del_targets.append(u)
if len(del_targets) > 0:
print('This will affect a total of %d users' % len(del_targets))
print('... waiting 5 seconds - Ctrl-C to abort! ...')
time.sleep(5)
print('This will affect a total of %d users, listing top 5:' % len(del_targets))
print('---------------------------------------------------------------')
for u in del_targets[:5]:
print(','.join([u.username, u.fullname or '', u.email, u.webpage_url or '']))
print('---------------------------------------------------------------')
print('... waiting 7 seconds - Ctrl-C to abort! ...')
time.sleep(7)
print('---------------------------------------------------------------')
print('username,fullname,email,webpage_url')
for u in del_targets:
print(','.join([u.username, u.fullname or '', u.email, u.webpage_url or '']))
if delete:
u.delete()
try:
u.delete()
except ForeignKeyViolation as ex:
print(ex.message)
else:
u.active = False
u.save()
print(','.join([u.username, u.fullname or '', u.email, u.webpage_url or '']))
delcount = delcount + 1
print('---------------------------------------------------------------')