def add_user_in_db(update, session): user = update.message.from_user username = getattr(user, 'username', sqlalchemy.null()) session.add(User(telegram_username=username, telegram_user_id=user.id, first_name=user.first_name, last_name=user.last_name, date_registration=datetime.datetime.now(), last_activity=datetime.datetime.now())) session.commit()
def create_server(dser): server_sql = q(Server).filter_by(id=str(dser.id)).first() if server_sql: print("Server already exists in the database") return None new_server = Server(id=str(dser.id)) session.add(new_server) session.commit() return new_server
def create_member(dmem): user_sql = q(User).filter_by(id=str(dmem.id)).first() if user_sql: print("Use already exists in database") return None new_user = User(id=str(dmem.id)) session.add(new_user) session.commit() return new_user
def create_membership(dmem, dser): membership_sql = q(Membership).filter_by(user_id=str(dmem.id), server_id=str(dser.id)).first() if membership_sql: print("Membership already exists in the database") return None new_membership = Membership(user_id=str(dmem.id), server_id=str(dser.id)) session.add(new_membership) session.commit() return new_membership
async def shorturl(self, ctx, url: str): """Shorten the given URL, the returned URL will redirect to the given URL.""" if len(url) > 2000: raise URLTooLarge() url_split = url.split(".") if not url.startswith("http://") and not url.startswith("https://"): url = "http://" + url if len(url_split) > 1: short = get_unique_short() new_shorturl = ShortUrl(location=url, short=short) session.add(new_shorturl) session.commit() await ctx.send( f":white_check_mark: URL shortened to https://www.oakchair.xyz/s/{short}" ) else: raise InvalidUrl()
def add_user(user_id): new_user = User(id=user_id, balance=0, level=1, xp=0) session.add(new_user) session.commit()
def add_server(guild_id): new_server = Server(id=guild_id) session.add(new_server) session.commit()