paulmataruso/open5gs
1
0
Fork 0
mirror of https://github.com/open5gs/open5gs.git synced 2025-11-15 03:11:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename dbconf.sh to open5gs-dbctl (#388)

Browse Source
This commit is contained in:
Sukchan Lee
2020-03-22 16:44:45 -04:00
parent 3d6a84ca85
commit db13836426
11 changed files with 69 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
misc/db/python/Open5GS.py Normal file
View File

@@ -0,0 +1,64 @@
import mongo
import pymongo
import random
import bson
class Open5GS:
def __init__(self, server, port):
self.server = server
self.port = port
def GetSubscribers(self):
myclient = pymongo.MongoClient("mongodb://" + str(self.server) + ":" + str(self.port) + "/")
mydb = myclient["open5gs"]
mycol = mydb["subscribers"]
subs_list = []
for x in mycol.find():
print(x)
subs_list.append(x)
pass
return subs_list
def GetSubscriber(self, imsi):
myclient = pymongo.MongoClient("mongodb://" + str(self.server) + ":" + str(self.port) + "/")
mydb = myclient["open5gs"]
mycol = mydb["subscribers"]
myquery = { "imsi": str(imsi)}
mydoc = mycol.find(myquery)
for x in mydoc:
print(x)
return x
def AddSubscriber(self, sub_data):
myclient = pymongo.MongoClient("mongodb://" + str(self.server) + ":" + str(self.port) + "/")
mydb = myclient["open5gs"]
mycol = mydb["subscribers"]
x = mycol.insert_one(sub_data)
print("Added subscriber with Inserted ID : " + str(x.inserted_id))
return x.inserted_id
def UpdateSubscriber(self, imsi, sub_data):
myclient = pymongo.MongoClient("mongodb://" + str(self.server) + ":" + str(self.port) + "/")
mydb = myclient["open5gs"]
mycol = mydb["subscribers"]
print("Attempting to update IMSI " + str(imsi))
newvalues = { "$set": sub_data }
myquery = { "imsi": str(imsi)}
x = mycol.update_one(myquery, newvalues)
print(x)
return True
def DeleteSubscriber(self, imsi):
myclient = pymongo.MongoClient("mongodb://" + str(self.server) + ":" + str(self.port) + "/")
mydb = myclient["open5gs"]
mycol = mydb["subscribers"]
myquery = { "imsi": str(imsi)}
x = mycol.delete_many(myquery)
print(x.deleted_count, " subscribers deleted.")
return x.deleted_count