Beispiel #1
0
async def shutdown(loop=None):
    loop = loop if loop is not None else asyncio.get_event_loop()
    pending = cancel_tasks()
    cprint(f'Cancelling {len(pending)} tasks...', color='yellow')
    with contextlib.suppress(asyncio.CancelledError):
        await asyncio.gather(*pending, return_exceptions=False)
    await asyncio.sleep(0)
Beispiel #2
0
    async def post_join_pk(self,
                           post_data: dict,
                           account: Optional[Account] = None):
        if account is None or account.banned:
            return

        url = 'https://api.live.bilibili.com/xlive/lottery-interface/v2/pk/join'
        headers = self.web_headers.copy()
        cookies = account.web_cookies.copy()
        for _ in range(3):
            response = await self.rl_session.post(url,
                                                  headers=headers,
                                                  cookies=cookies,
                                                  data=post_data)
            try:
                response.raise_for_status()
                js = await response.json()

                if 0 == js['code']:
                    return js
                elif '访问被拒绝' == js.get('message'):
                    account.banned = True
                    account.cprint(f'访问被拒绝', color='purple')
                    return None
                elif '领取过' in js.get('message'):
                    account.cprint(f'已领取', color='purple')
                    return None
                else:
                    return None
            except (aiohttp.ClientPayloadError,
                    aiohttp.ClientResponseError) as exc:
                cprint(f'aiohttp ClientError: {exc}', color='red')
Beispiel #3
0
    def importdb(conf, dbfile, passwd, passphrase):
        if PasswdManager.verbose:
            cprint('Reading and decrypting database', color='blue')
        g = Giltzarrapo().readEncrypted(conf['dbfile']).decrypt(
            passwd, conf['privkey'], passphrase)

        if PasswdManager.verbose:
            print(cprepare('Reading plain exported file :', color='blue'),
                  cprepare(dbfile, color='lblue'))
        content = InternalPasswdManager.readfile(dbfile)
        header_fields = ['service', 'user', 'password']
        old_header, lines = content[0].split('\t'), [
            l.split('\t') for l in content[1:]
        ]
        line_template = '{}\n'.format('\t'.join([
            ('{}' if h in header_fields else ' ') for h in old_header
        ]))

        if PasswdManager.verbose:
            print(cprepare('Inserting rows in', color='blue'),
                  cprepare(conf['dbfile'].split('/')[-1], color='lblue'),
                  cprepare('database', color='blue'))
        for l in lines:
            InternalPasswdManager.append(g, line_template.format(*l))

        if PasswdManager.verbose:
            cprint('Saving modified database', color='blue')
        g.encrypt(passwd, conf['pubkey']).save(conf['dbfile'])
Beispiel #4
0
 def process_login(self):
     accounts = [
         Account(data_file, self.yaml) for data_file in self.account_files
     ]
     self.accounts = accounts
     wait = False
     for acc in accounts:
         if not acc.usable and acc.username and acc.password:
             l = Login(acc, output_file=acc.input_file, yaml=self.yaml)
             try:
                 cprint(f'{acc.input_file!r}: Logging in...', color='green')
                 l.login()
                 acc.reload()
                 acc.cprint(f'登录成功', color='green')
                 if wait:
                     time.sleep(5)
             except LoginException as exc:
                 acc.cprint(f'登录失败 - {exc}', error=True)
             wait = True
         elif acc.usable:
             msg = (f'{acc.input_file!r}: cookies读取成功')
             acc.cprint(f'{msg}', color='green')
         else:
             msg = (f'登录信息不完整 && cookies未提供. - '
                    f' {acc.input_file!r} Skipped.')
             cprint(f'{msg}', color='yellow')
Beispiel #5
0
    async def post(self,
                   url,
                   params=None,
                   headers=None,
                   cookies=None,
                   data=None,
                   rate_limited: bool = True):
        if rate_limited == True:
            session = self.rl_session
        else:
            session = self.session

        err = None
        for _ in range(3):
            # fire request
            try:
                response = await session.post(url,
                                              params=params,
                                              headers=headers,
                                              cookies=cookies,
                                              data=data,
                                              ssl=False)

                response.raise_for_status()

                r_json = await response.json(content_type=None)

                return r_json
            except aiohttp.ClientError as exc:
                err = exc
                cprint(f'aiohttp ClientError: {exc}', color='red')

        if err:
            raise HttpError(f'Request failed: {err} @url {url!r}')
Beispiel #6
0
    def create(dbname, passwd, passphrase, databases_path, keys_path):
        if PasswdManager.verbose:
            print(cprepare('Creating', color='blue'),
                  cprepare(dbname, color='lblue'),
                  cprepare('database', color='blue'))
        g = Giltzarrapo()
        InternalPasswdManager.append(
            g,
            'passranoid_database:v1.1\tpassranoid_interface:v1.2\tpassranoid_config:v1\n'
        )
        g.status = 'plain'

        if PasswdManager.verbose:
            cprint('Generating rsa key pair', color='blue')
        privkey, pubkey = Giltzarrapo.generateRSApair(passphrase,
                                                      dir=keys_path,
                                                      name=dbname)
        if PasswdManager.verbose:
            print(cprepare('Keys saved at :\n  - Privkey :', color='blue'),
                  cprepare(privkey, color='lblue'),
                  cprepare('\n  - Pubkey :', color='blue'),
                  cprepare(pubkey, color='lblue'))

        if PasswdManager.verbose:
            print(
                cprepare('Saving database at :', color='blue'),
                cprepare('{}/{}'.format(databases_path, dbname),
                         color='lblue'))
        g.encrypt(passwd, pubkey).save('{}/{}'.format(databases_path, dbname))
Beispiel #7
0
 def usable(self):
     usable = False
     try:
         usable = self.web_cookies and all(
             [cookie for cookie in self.web_cookies.values()])
     except AttributeError:
         cprint(f'AttributeError in account.py', error=True)
     return usable
Beispiel #8
0
 def print_time_elapsed(self):
     elapsed = days_to_date(self.time_elapsed)
     years = elapsed[0]
     months = elapsed[1]
     days = elapsed[2]
     cprint('It has been %s year(s), %s month(s), and %s day(s) since you began your search. '
            % (years, months, days))
     return 0
Beispiel #9
0
    def list_room_contents(self, room):
        object_dic = None

        for e in self.rooms_desc:
            if e.lower() == room.lower():
                object_dic = self.rooms_objs[e]

        for obj in object_dic:
            cprint('\t' + obj)
Beispiel #10
0
async def get_area_of_room(roomid: int):
    response = await Bilibili().request_check_live_info(roomid)
    if response is None:
        return
    if response['code'] == 0:
        return response['data']['parent_area_id']
    else:
        cprint(f'Get area of room status code {response["code"]}', 
               color='red')
Beispiel #11
0
    def changedbpass(conf, newpasswd, passwd, passphrase):
        if PasswdManager.verbose:
            cprint('Reading and decrypting database', color='blue')
        g = Giltzarrapo().readEncrypted(conf['dbfile']).decrypt(
            passwd, conf['privkey'], passphrase)

        if PasswdManager.verbose:
            cprint('Encrypting database with the new password', color='blue')
        g.encrypt(newpasswd, conf['pubkey']).save(conf['dbfile'])
Beispiel #12
0
async def get_all_rooms_on_page(page:int, 
                                page_size:int = 99, 
                                popularity_over:int = 50):
    response = await Bilibili().request_check_area(0, page=page, page_size=page_size)
    if response['code'] != 0:
        cprint(f'Status code in get_all_rooms_on_page '
               f'{response["code"]} {response["message"]}', color='red')
        raise RuntimeError('Status code nonzero')
    for room_info in response['data']['list']:
        if room_info['online'] > popularity_over:
            yield room_info['roomid']
Beispiel #13
0
 def verifyauth(conf, passwd, passphrase):
     if PasswdManager.verbose:
         cprint('Verifying database credentials', color='blue')
     try:
         g = Giltzarrapo().readEncrypted(conf['dbfile']).decrypt(
             passwd, conf['privkey'], passphrase)
         g.encrypt(passwd, conf['pubkey']).save(conf['dbfile'])
     except:
         return False
     if PasswdManager.verbose:
         cprint('Successfully verification', color='green')
     return True
 def on_raffle(self, data):
     t = ''
     if data['type'] == 'guard':
         t = 'guard'
     elif data['type'] == 'pk':
         t = 'pk'
     elif data['type'] == 'storm':
         t = 'storm'
     else:
         t = 'gift'
     
     self.emitter and self.emitter.emit(t, data)
     cprint(f'{data["id"]:<13} @{data["roomid"]:<12} {data["name"]}', 
         color='cyan')
Beispiel #15
0
    def holidays_passed(self, time_elapsed):
        time = self.date
        had_christmas = False
        had_birthday = False
        had_new_year = False
        had_valentines = False

        while time_elapsed != 0:
            time += timedelta(1)
            if time.month == self.christmas[1] and time.day == self.christmas[0]:
                had_christmas = True
            elif time.month == self.birthday[1] and time.day == self.birthday[0]:
                had_birthday = True
            elif time.month == self.new_year[1] and time.day == self.new_year[0]:
                had_new_year = True
            elif time.month == self.valentines[1] and time.day == self.valentines[0]:
                had_valentines = True
            time_elapsed -= 1

        if had_birthday:
            cprint('Your birthday has passed. ', 0.15)
            self.birthdays_had += 1
        if had_christmas:
            cprint('Christmas has passed. ', 0.15)
        if had_new_year:
            cprint('A new year has come. ', 0.15)
        if had_valentines:
            cprint("Valentine's has passed. ", 0.15)

        return 0
Beispiel #16
0
    def exportdb(conf, dbfile, passwd, passphrase):
        if PasswdManager.verbose:
            cprint('Reading and decrypting database', color='blue')
        g = Giltzarrapo().readEncrypted(conf['dbfile']).decrypt(
            passwd, conf['privkey'], passphrase)

        if PasswdManager.verbose: cprint('Fetching rows', color='blue')
        header = '\t'.join(['service', 'user', 'password'])
        lines = [l for i, l in InternalPasswdManager.getlines(g, '*')]
        if len(lines) == 0: return False

        if PasswdManager.verbose:
            print(cprepare('Saving plain database rows at :', color='blue'),
                  cprepare(dbfile, color='lblue'))
        InternalPasswdManager.savefile(header=header, lines=lines, file=dbfile)
        return True
Beispiel #17
0
    def reload(self):
        data_file = utils.resolve_to_cwd(self.input_file)
        try:
            with open(data_file, mode='r', encoding='UTF-8') as iofile:
                store = self.yaml.load(iofile)

            self.username = store.get('user').get('username')
            self.password = store.get('user').get('password')
            self.web_cookies = store.get('web')
            self.app_cookies = store.get('app')
            try:
                self.uid = self.web_cookies.get('DedeUserID')
            except AttributeError:
                pass
        except FileNotFoundError:
            msg = (f'File {self.input_file!r} not found. Skipped')
            cprint(f'{msg}', error=True)
        except (TypeError, AttributeError):
            pass
    async def connect(self):
        cls = self.__class__
        url = self.server["url"] or f'ws://{self.server["address"]}:{self.server["port"]}'
        async with websockets.connect(url) as ws:
            self.connected = True
            self.ws = ws
            accepted = False
            try:
                await ws.send(self.handshake)

                cprint(f'Established connection with [ {ws.remote_address} ]', color='green')
                accepted = True

                while accepted and ws.open:
                    data = await ws.recv()
                    body = self.deserialize(data)
                    self.on_raffle(body)

                await ws.close()
                await ws.wait_closed()
                self.accepted = False
            except websockets.ConnectionClosed:
                if accepted:
                    cprint(f'Lost connection: [ {ws.remote_address} ]', color='red')
                self.again = True
            except IndexError:
                pass
            except Exception as exc:
                cprint(f'{type(exc)}: {exc}', error=True)
                self.again = False
            finally:
                self.connected = False
Beispiel #19
0
    def remove(conf, index, passwd, passphrase):
        if PasswdManager.verbose:
            cprint('Reading and decrypting database', color='blue')
        g = Giltzarrapo().readEncrypted(conf['dbfile']).decrypt(
            passwd, conf['privkey'], passphrase)

        if PasswdManager.verbose:
            cprint('Removing matching rows', color='blue')
        if index == '*':
            removerows = [
                l[1].encode('utf-8')
                for l in InternalPasswdManager.getlines(g, '*')
            ]
        else:
            try:
                removerows = InternalPasswdManager.getlines(
                    g, '*')[index][1].encode('utf-8')
            except IndexError:
                return False
        InternalPasswdManager.remove(g, removerows)

        if PasswdManager.verbose:
            cprint('Saving modified database', color='blue')
        g.encrypt(passwd, conf['pubkey']).save(conf['dbfile'])
        return True
Beispiel #20
0
    def changedbkey(conf, newpassphrase, passwd, passphrase):
        if PasswdManager.verbose:
            cprint('Reading and decrypting database', color='blue')
        g = Giltzarrapo().readEncrypted(conf['dbfile']).decrypt(
            passwd, conf['privkey'], passphrase)

        if PasswdManager.verbose:
            cprint('Generating rsa key pair', color='blue')
        keys_path, keyname = '/'.join(
            conf['privkey'].split('/')[:-1]), conf['privkey'].split('/')[-1]
        os.rename(conf['privkey'], '{}/.old.{}'.format(keys_path, keyname))
        os.rename(conf['pubkey'], '{}/.old.{}.pub'.format(keys_path, keyname))
        privkey, pubkey = Giltzarrapo.generateRSApair(newpassphrase,
                                                      dir=keys_path,
                                                      name=keyname)
        if PasswdManager.verbose:
            print(
                cprepare('Old keys moved to :\n  - Privkey :', color='blue'),
                cprepare('{}/.old.{}'.format(keys_path, keyname),
                         color='lblue'),
                cprepare('\n  - Pubkey :', color='blue'),
                cprepare('{}/.old.{}.pub'.format(keys_path, keyname),
                         color='lblue'))
        if PasswdManager.verbose:
            print(cprepare('New keys saved at :\n  - Privkey :', color='blue'),
                  cprepare(privkey, color='lblue'),
                  cprepare('\n  - Pubkey :', color='blue'),
                  cprepare(pubkey, color='lblue'))

        if PasswdManager.verbose:
            cprint('Encrypting database with the new key pair', color='blue')
        g.encrypt(passwd, pubkey).save(conf['dbfile'])
Beispiel #21
0
def login(username, password):
    response = login_simple(username, password)

    # raise errors
    if response.status_code != 200:
        raise LoginException(f'LoginError: ' + response.status_code)
    r_json = response.json()
    if r_json['code'] != 0:
        cprint(r_json['message'], color='red')
        cprint('登录失败(返回值非0)', color='red')
        cprint('解决方法:重启程序再试(建议仅重试一次)', color='green')
        raise LoginException(f'LoginError')

    # handle cookies
    cookies = {}
    cookie_source = [cookie for cookie in r_json['data']['cookie_info']['cookies']]
    for cookie in cookie_source:
        cookies[cookie['name']] = cookie['value']

    # handle tokens
    tokens = {}
    token_source = { key:value for key,value in r_json['data']['token_info'].items() } 
    tokens.update(token_source)

    return { 'app': tokens, 'web': cookies, }
    async def run(self):
        while self.again and self.server != None:
            try:
                done, pending = await asyncio.wait(
                    [
                        self.timeout(), 
                        self.connect(), 
                    ], 
                    loop=self.loop, 
                    return_when=asyncio.FIRST_EXCEPTION, 
                )

                try:
                    for task in done:
                        if task.exception():
                            raise task.exception()

                    if not self.accepted:
                        msg = (f'ConnectionRefused: Password mismatch')
                        cprint(f'{msg}', error=True)
                        sys.exit(1)
                except (ConnectionRefusedError, TimeoutError, OSError):
                    await self.on_refused()
                except websockets.InvalidMessage:
                    pass
            except NoAvailableServer:
                cprint(f'NoAvailableServer: Stop connecting.', error=True)
                self.again = False
                self.server = None
            finally:
                for task in pending:
                    try:
                        task.cancel()
                        await task
                    except Exception:
                        pass
Beispiel #23
0
    def list(conf, passwd, passphrase):
        if PasswdManager.verbose:
            cprint('Reading and decrypting database', color='blue')
        g = Giltzarrapo().readEncrypted(conf['dbfile']).decrypt(
            passwd, conf['privkey'], passphrase)

        if PasswdManager.verbose: cprint('Fetching rows', color='blue')
        lines = InternalPasswdManager.getlines(g, '*')

        if PasswdManager.verbose: cprint('Refreshing database', color='blue')
        g.encrypt(passwd, conf['pubkey']).save(conf['dbfile'])
        return lines
Beispiel #24
0
    def version(conf, passwd, passphrase):
        if PasswdManager.verbose:
            cprint('Reading and decrypting database', color='blue')
        g = Giltzarrapo().readEncrypted(conf['dbfile']).decrypt(
            passwd, conf['privkey'], passphrase)

        if PasswdManager.verbose:
            cprint('Fetching database version', color='blue')
        version = InternalPasswdManager.getversion(g)

        if PasswdManager.verbose: cprint('Refreshing database', color='blue')
        g.encrypt(passwd, conf['pubkey']).save(conf['dbfile'])
        return version
Beispiel #25
0
    def introduce_room_contents(self, room):
        desc = None
        object_dic = None

        for e in self.rooms_desc:
            if e.lower() == room.lower():
                desc = self.rooms_desc[e]
                # print desc
                object_dic = self.rooms_objs[e]

        cprint(desc)
        cprint('You see the following: ')
        for obj in object_dic:
            cprint('\t' + obj)

        return object_dic
Beispiel #26
0
    def insert(conf, service, user, spasswd, passwd, passphrase):
        if PasswdManager.verbose:
            cprint('Reading and decrypting database', color='blue')
        g = Giltzarrapo().readEncrypted(conf['dbfile']).decrypt(
            passwd, conf['privkey'], passphrase)

        if PasswdManager.verbose: cprint('Inserting new row', color='blue')
        if spasswd == '': spasswd = PasswdManager.passgen()
        InternalPasswdManager.append(
            g, '{}\t{}\t{}\n'.format(service, user, spasswd))
        if PasswdManager.verbose:
            ctable(header=['service', 'user', 'password'],
                   data=[[service, user, spasswd]],
                   header_color='blue',
                   rows_color='lblue')

        if PasswdManager.verbose:
            cprint('Saving modified database', color='blue')
        g.encrypt(passwd, conf['pubkey']).save(conf['dbfile'])
Beispiel #27
0
 def change_date_dialogue(days):
     if days > 1:
         cprint("%s days have passed." % days)
     else:
         cprint("One day has passed.")
Beispiel #28
0
 def inspect_detailed(self):
     cprint(string=("ID: " + self.name), t=0.12)
     cprint(string='. . . . . ', t=0.30)
     sleep(0.2)
     return self.header + '\n' + self.desc_detailed + '\n' + self.footer
Beispiel #29
0
 def inspect_vague(self):
     cprint(string=("OBJ-ID: " + self.name), t=0.12)
     cprint(string='. . . . . ', t=0.30)
     sleep(0.2)
     return self.header + '\n' + self.desc_vague
Beispiel #30
0
 def error(text):
     cprint(text, color='red')
Beispiel #31
0
 def print_date(self):
     cprint('It is currently the year %s, day %s of month %s.'
            % (self.date.year, self.date.day, self.date.month), 0.02)
     return 0
 async def on_refused(self):
     msg = (f'ConnectionRefused: Server may be down. ')
     self.server = self.next_server()
     msg += f'(Switching to {self.server["address"]})'
     cprint(f'{msg}', color='yellow')
Beispiel #33
0
 def cprint(self, *args, color='default', **kwargs):
     prefix = f'[{self.uid}]'
     data = ' '.join(args)
     cprint(f'{prefix:<13} {data}', color=color, **kwargs)
Beispiel #34
0
 except:
     break
 else:
     command, args = command.split(' ')[0], command.split(' ')[1:]
     try:
         if command == 'create':
             create_args = cm.handle(command, args)
             if create_args != False: PasswdManager.create(*create_args)
         elif command == 'import':
             import_args = cm.handle(command, args)
             if import_args != False: PasswdManager.importdb(*import_args)
         elif command == 'passgen':
             passgen_args = cm.handle(command, args)
             if passgen_args != False:
                 newpasswd = PasswdManager.passgen(*passgen_args)
                 cprint('Generated password : '******'blue', end="")
                 cprint(newpasswd, color='lblue')
         elif command == 'clear':
             clear()
         elif command == 'help':
             print_help()
         elif command == 'use':
             verify_args = cm.handle(command, args)
             if verify_args != False:
                 success = PasswdManager.verifyauth(*verify_args)
                 if not success:
                     cprint('Wrong password or/and passphrase', color='red')
                 else:
                     cm.set_auth(*verify_args)
         elif command == 'insert' or command == 'add':
             insert_args = cm.handle('insert', args)
Beispiel #35
0
import sys
from giltzarrapo import Giltzarrapo
from printer import cprint, ecprint
from datetime import datetime as dt

passph = '123'
passwd = 'abc'
num_process = 3
sb = None

if '-e' in sys.argv:
    #RSA KEYS
    cprint('Generando claves...', color='cyan', end='\r')
    s = dt.now()
    Giltzarrapo.generateRSApair(passphrase=passph)
    e = dt.now()
    ecprint(['Generacion de claves ', str(e - s)],
            color=['cyan', 'yellow'],
            template='{} : {}')

    #ENCRYPT
    g = Giltzarrapo(n_processes=num_process)
    #read
    cprint('Leyendo fichero...', color='cyan', end='\r')
    s = dt.now()
    g.readPlain('testfile.txt')
    e = dt.now()
    ecprint(['Lectura de fichero   ', str(e - s)],
            color=['cyan', 'yellow'],
            template='{} : {}')
    #encrypt