Coverage for procpath/cmd/explore.py: 100%

1import logging (empty)

2import os (empty)

3import shutil (empty)

4import threading (empty)

5import webbrowser (empty)

6from functools import partial (empty)

7from pathlib import Path (empty)

8from typing import Optional (empty)

9from urllib.parse import urlencode (empty)

10 

11from .. import sqliteviz (empty)

12 

13 

14__all__ = 'run', (empty)

15 

16logger = logging.getLogger('procpath') (empty)

17 

18 

19def run( (empty)

20 bind: str, 

21 port: int, 

22 open_in_browser: bool, 

23 reinstall: bool, 

24 build_url: str, 

25 database_file: Optional[str] = None, 

26): 

27 user_cache_dir = Path(os.getenv('XDG_CACHE_HOME', os.path.expanduser('~/.cache'))) 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

28 sqliteviz_dir = user_cache_dir / 'procpath' / 'sqliteviz' 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

29 if not sqliteviz_dir.exists() or reinstall: 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

30 shutil.rmtree(sqliteviz_dir, ignore_errors=True) 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

31 sqliteviz_dir.mkdir(parents=True) 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

32 logger.info('Downloading %s into %s', build_url, sqliteviz_dir) 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

33 sqliteviz.install_sqliteviz(build_url, sqliteviz_dir) 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

34 else: 

35 logger.info('Serving existing Sqliteviz from %s', sqliteviz_dir) 1 ctxprocpath.test.cmd.TestExploreCommand.test_explore

36 

37 url = 'http://{host}:{port}/'.format(port=port, host=bind or 'localhost') 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

38 logger.info('Serving Sqliteviz at %s', url) 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

39 

40 server_fn = partial(sqliteviz.serve_dir, bind, port, str(sqliteviz_dir)) 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

41 server = threading.Thread(target=server_fn, daemon=True) 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

42 server.start() 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

43 

44 if database_file: 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

45 try: 2 ctxprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

46 sym_path = sqliteviz.symlink_database(database_file, sqliteviz_dir) 2 ctxprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

47 except FileNotFoundError: 1 ctxprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

48 logger.warning('Database file %s does not exist', database_file) 1 ctxprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

49 else: 

50 params = urlencode({'data_url': url + sym_path.name, 'data_format': 'sqlite'}) 1 ctxprocpath.test.cmd.TestExploreCommand.test_explore_preload_database

51 url += f'#/load?{params}' 1 ctxprocpath.test.cmd.TestExploreCommand.test_explore_preload_database

52 

53 if open_in_browser: 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

54 webbrowser.open(url) 3 ctxprocpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing

55 

56 server.join() 4 ctx(empty)procpath.test.cmd.TestExploreCommand.test_exploreprocpath.test.cmd.TestExploreCommand.test_explore_preload_databaseprocpath.test.cmd.TestExploreCommand.test_explore_preload_database_missing