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