33 lines
734 B
Python
33 lines
734 B
Python
|
import sys
|
||
|
|
|
||
|
|
from decouple import config
|
||
|
|
|
||
|
|
from ereuse_devicehub.db import db
|
||
|
|
from ereuse_devicehub.devicehub import Devicehub
|
||
|
|
from ereuse_devicehub.modules.oidc.models import MemberFederated
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
"""
|
||
|
|
We need add client_id and client_secret for every server
|
||
|
|
than we want connect.
|
||
|
|
"""
|
||
|
|
schema = config('DB_SCHEMA')
|
||
|
|
app = Devicehub(inventory=schema)
|
||
|
|
app.app_context().push()
|
||
|
|
domain = sys.argv[1]
|
||
|
|
client_id = sys.argv[2]
|
||
|
|
client_secret = sys.argv[3]
|
||
|
|
member = MemberFederated.query.filter_by(domain=domain).first()
|
||
|
|
if not member:
|
||
|
|
return
|
||
|
|
|
||
|
|
member.client_id = client_id
|
||
|
|
member.client_secret = client_secret
|
||
|
|
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|