28 lines
631 B
Python
28 lines
631 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():
|
||
|
|
schema = config('DB_SCHEMA')
|
||
|
|
app = Devicehub(inventory=schema)
|
||
|
|
app.app_context().push()
|
||
|
|
dlt_id_provider = sys.argv[1]
|
||
|
|
domain = sys.argv[2]
|
||
|
|
member = MemberFederated.query.filter_by(domain=domain).first()
|
||
|
|
if member:
|
||
|
|
return
|
||
|
|
|
||
|
|
member = MemberFederated(domain=domain, dlt_id_provider=dlt_id_provider)
|
||
|
|
|
||
|
|
db.session.add(member)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|