def connection(name=None): global _db if not _db: _db = MongoClient('mongodb://172.17.1.201:27017') return _db
async def count(self): """Проверяет количество объектов в базе с такими gplay_id и hl""" client = MongoClient(connectstr) db = client['gplay-permissions'] records = db.records query = {"name": self.gplay_id, "hl": self.hl} ret = await records.count_documents(query) return ret
async def read(self): """Считывает даннные из MongoDB и возвращает Permissions""" client = MongoClient(connectstr) db = client['gplay-permissions'] records = db.records query = {"name": self.gplay_id, "hl": self.hl} if await records.count_documents(query) == 0: await self.parse() await self.save() ret = await records.find_one(query) return ret
async def save(self): """Сохраняет полученные данные из self.permissions в MongoDB""" client = MongoClient(connectstr) db = client['gplay-permissions'] records = db.records data = { "name": self.gplay_id, "hl": self.hl, "permissions": self.permissions } await records.insert_one(data)
def __init__(self, config): self.client = MongoClient(config.mongo.host, config.mongo.port, username=config.mongo.username, password=config.mongo.password) self.bot_data = self.client.bot_data self.config = self.bot_data.config self.cmd_history = self.bot_data.cmd_history self.stats = self.bot_data.stats self.bot_guilds = self.bot_data.bot_guilds self.plugin_data = self.client.plugin_data log.debug( f'connected to mongo instance at {config.mongo.host}:{config.mongo.port}' )
DEV_USERS.add(1037581197) if not SPAMWATCH_API: sw = None LOGGER.warning("SpamWatch API key missing! recheck your config.") else: try: sw = spamwatch.Client(SPAMWATCH_API) except: sw = None LOGGER.warning("Can't connect to SpamWatch!") updater = tg.Updater(TOKEN, workers=WORKERS, use_context=True) telethn = TelegramClient("masha", API_ID, API_HASH) pbot = Client("mashapbot", api_id=API_ID, api_hash=API_HASH, bot_token=TOKEN) mongo_client = MongoClient(MONGO_DB_URI) db = mongo_client.SaitamaRobot dispatcher = updater.dispatcher DRAGONS = list(DRAGONS) + list(DEV_USERS) DEV_USERS = list(DEV_USERS) WOLVES = list(WOLVES) DEMONS = list(DEMONS) TIGERS = list(TIGERS) # Load at end to ensure all prev variables have been set from MashaRoBot.modules.helper_funcs.handlers import ( CustomCommandHandler, CustomMessageHandler, CustomRegexHandler, )
# Support Dual Mongo DB now # For free users from motor.motor_asyncio import AsyncIOMotorClient as MongoClient from AuraX.config import get_str_key MONGO2 = get_str_key("MONGO_URI_2", None) MONGO = get_str_key("MONGO_URI", required=True) if MONGO2 == None: MONGO2 = MONGO mongo_client = MongoClient(MONGO2) db = mongo_client.AuraXRobot
log = logging.getLogger(__name__) log.setLevel(logging.DEBUG) proxy_adapter = { 'connector': ProxyConnector(remote_resolve=True), 'request_class': ProxyClientRequest } proxies = { 'http': 'socks5://127.0.0.1:1080', 'https': 'socks5://127.0.0.1:1080' } loop = asyncio.get_event_loop() m = MongoClient(settings.MONGO) pool = Pool(50, loop) def get_address(addr, name): _addr = '' if addr['address1']: _addr += addr['address1'] else: _addr += name if addr['address2'] and addr['address2'] != addr['city']: _addr += ', ' + addr['address2'] if addr['address3'] and addr['address3'] != addr['city']: _addr += ', ' + addr['address3'] if addr['state']:
from aioutils import Pool from aiohttp import ClientSession from googletrans import Translator from motor.motor_asyncio import AsyncIOMotorClient as MongoClient from scripture.utils import strip_tags from scripture.utils.jsetadditional import JsetAdditional # os.environ['no_proxy'] = '47.94.77.75' # os.environ['http_proxy'] = 'socks5://127.0.0.1:1080' # os.environ['https_proxy'] = 'socks5://127.0.0.1:1080' asyncio.set_event_loop_policy(uvloop.EventLoopPolicy()) m = MongoClient('mongodb://127.0.0.1:27017') t = Translator(service_urls=['translate.google.cn']) _redis = redis.Redis() logging.basicConfig( format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.ERROR) log = logging.getLogger('hotel') cities = [ 'Amsterdam', 'Barcelona', 'Berlin', 'Boston',
else: from sample_config import * USERBOT_PREFIX = USERBOT_PREFIX GBAN_LOG_GROUP_ID = GBAN_LOG_GROUP_ID SUDOERS = SUDO_USERS_ID WELCOME_DELAY_KICK_SEC = WELCOME_DELAY_KICK_SEC LOG_GROUP_ID = LOG_GROUP_ID MESSAGE_DUMP_CHAT = MESSAGE_DUMP_CHAT MOD_LOAD = [] MOD_NOLOAD = [] bot_start_time = time.time() # MongoDB client print("[INFO]: INITIALIZING DATABASE") mongo_client = MongoClient(MONGO_URL) db = mongo_client.wbb async def load_sudoers(): global SUDOERS print("[INFO]: LOADING SUDOERS") sudoersdb = db.sudoers sudoers = await sudoersdb.find_one({"sudo": "sudo"}) sudoers = [] if not sudoers else sudoers["sudoers"] for user_id in SUDOERS: if user_id not in sudoers: sudoers.append(user_id) await sudoersdb.update_one( {"sudo": "sudo"}, {"$set": {
def databases(name): if name not in _db_cache: connection = MongoClient(settings.DATABASES[name]) _db_cache[name] = connection.get_database() return _db_cache[name]
def before_server_start(app, loop): from motor.motor_asyncio import AsyncIOMotorClient as MongoClient MONGO_URI = app.config.get('MONGO') cls.set_connection(MongoClient(MONGO_URI, io_loop=loop), loop)
def __init__(self, bot): self.bot = bot self.mongo_users = MongoClient(getenv('MONGO_URI')).popskill.user_links self.token = None self.watch.start()
}}) async def ban_user(self, user_id, ban_duration, ban_reason): ban_status = dict(is_banned=True, ban_duration=ban_duration, banned_on=datetime.date.today().isoformat(), ban_reason=ban_reason) await self.col.update_one({'id': user_id}, {'$set': { 'ban_status': ban_status }}) async def get_ban_status(self, id): default = dict(is_banned=False, ban_duration=0, banned_on=datetime.date.max.isoformat(), ban_reason='') user = await self.col.find_one({'id': int(id)}) return user.get('ban_status', default) async def get_all_banned_users(self): banned_users = self.col.find({'ban_status.is_banned': True}) return banned_users # Database db = Database(DATABASE_URL, BOT_USERNAME) # Database for Anti-Cmd (I know what you are thinking about! but IDC) mongo_db_lmao = MongoClient(DATABASE_URL) dcmdb = mongo_db_lmao.handlers