k6407399 commented on issue #3083:
URL: https://github.com/apache/superset/issues/3083#issuecomment-1475561246
> @SWoto code really helped me a lot. It didn't work right away for my needs:
>
> * delete dozens of users read from a .jsonl file
> * add dozens of users read from a .jsonl file
>
> But after slightly modifying it, I could do all of that programmatically:
>
> ```
> import requests
> from bs4 import BeautifulSoup as bs
> from bs4 import Comment
> import json
>
> class UserSupersetApi:
> def __init__(self, username=None, password=None):
> self.s = requests.Session()
> self.base_url = "superset.xyz"
> self._csrf = self._getCSRF(self.url('login/'))
> self.headers = {'X-CSRFToken': self._csrf, 'Referer':
self.url('login/')}
> # note: does not use headers because of
flask_wtf.csrf.validate_csrf
> # if data is dict it is used as form and ends up empty but
flask_wtf checks if data ...
> self.s.post(self.url('login/'),
> data={'username': username, 'password': password, 'csrf_token':
self._csrf},
> allow_redirects=False)
>
> def url(self, url_path):
> return self.base_url + url_path
>
> def get(self, url_path):
> return self.s.get(self.url(url_path), headers=self.headers)
>
> def post(self, url_path, data=None, json_data=None, **kwargs):
> kwargs.update({'url': self.url(url_path), 'headers': self.headers})
> if data:
> data['csrf_token'] = self._csrf
> kwargs['data'] = data
> if json_data:
> kwargs['json'] = json_data
> return self.s.post(**kwargs, allow_redirects=False)
>
> def _getCSRF(self, url_path):
> response = self.s.get(url_path)
> soup = bs(response.content, "html.parser")
> for tag in soup.find_all('input', id='csrf_token'):
> csrf_token = tag['value']
> return csrf_token
>
> '''
> builds User Name
> fails if 2 people have the same name
> '''
> @staticmethod
> def build_username (first_name, last_name):
> return f'{first_name}_{last_name}'
>
> ## authenticate + set global vars
>
> superset_username = 'ff'
> superset_pw = 'fff'
> file_name = 'students_data.jsonl'
>
> superset = UserSupersetApi(superset_username, superset_pw)
>
> ## create users based on file
>
> # users = [
> # {
> # 'first_name': 'one',
> # 'last_name':'one',
> # 'username': 'one',
> # 'email': '[email protected]',
> # 'password': 'one',
> # 'conf_password': 'one',
> # 'roles': []
> # },
> # {
> # 'first_name': 'two',
> # 'last_name':'two',
> # 'username': 'two',
> # 'email': '[email protected]',
> # 'password': 'two',
> # 'conf_password': 'two',
> # 'roles': []
> # }
> # ]
> # for user in users:
> # payload = {'first_name': user['first_name'],
> # 'last_name': user['last_name'],
> # 'username': user['username'],
> # 'email': user['email'],
> # 'active': True,
> # 'conf_password': user['password'],
> # 'password': user['password'],
> # 'roles': user['roles']}
> # print(superset.post(url_path='users/add', json=payload))
>
> with open(file_name , 'r') as f:
> for l in f:
> user_dct = json.loads(l)
>
>
> payload = {'first_name': user_dct['first_name'],
> 'last_name': user_dct['last_name'],
> 'username':
UserSupersetApi.build_username(user_dct['first_name'], user_dct['last_name']),
> 'email': user_dct['email'],
> 'active': True,
> 'conf_password': user_dct['password'],
> 'password': user_dct['password'],
> 'roles': user_dct['roles']
> }
>
> superset.post(url_path=f'users/add', json=payload)
>
> ## delete users based on file
>
> # provides pk given a username
> res = superset.get(url_path='users/api/read')
>
> pks = res.json()['pks']
> users = res.json()['result']
> n_pk = len(pks)
>
> # build dict of all current users
> pk_lut = dict()
> for i in range(n_pk):
> pk_lut[users[i]['email']] = pks[i]
>
> # for all entries in file delete users from superset based on email
> with open(file_name , 'r') as f:
> for l in f:
> user_dct = json.loads(l)
> pk = pk_lut[user_dct['email']]
> superset.post(url_path=f'users/delete/{pk}')
> ```
>
> Hope this helps someone.
>
> Best regards Tridelt
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
