FilesExpand file tree

 master

/

patch_linecache.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

75 lines (64 loc) · 2.69 KB

 master

/

patch_linecache.py

Copy path

Top

File metadata and controls

• Code
• Blame

75 lines (64 loc) · 2.69 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

import linecache

class BPythonLinecache(dict):

"""Replaces the cache dict in the standard-library linecache module,

to also remember (in an unerasable way) bpython console input."""

def __init__(self, *args, **kwargs):

super(BPythonLinecache, self).__init__(*args, **kwargs)

self.bpython_history = []

def is_bpython_filename(self, fname):

try:

return fname.startswith('<bpython-input-')

except AttributeError:

# In case the key isn't a string

return False

def get_bpython_history(self, key):

"""Given a filename provided by remember_bpython_input,

returns the associated source string."""

try:

idx = int(key.split('-')[2][:-1])

return self.bpython_history[idx]

except (IndexError, ValueError):

raise KeyError

def remember_bpython_input(self, source):

"""Remembers a string of source code, and returns

a fake filename to use to retrieve it later."""

filename = '<bpython-input-%s>' % len(self.bpython_history)

self.bpython_history.append((len(source), None,

source.splitlines(True), filename))

return filename

def __getitem__(self, key):

if self.is_bpython_filename(key):

return self.get_bpython_history(key)

return super(BPythonLinecache, self).__getitem__(key)

def __contains__(self, key):

if self.is_bpython_filename(key):

try:

self.get_bpython_history(key)

return True

except KeyError:

return False

return super(BPythonLinecache, self).__contains__(key)

def __delitem__(self, key):

if not self.is_bpython_filename(key):

return super(BPythonLinecache, self).__delitem__(key)

def _bpython_clear_linecache():

try:

bpython_history = linecache.cache.bpython_history

except AttributeError:

bpython_history = []

linecache.cache = BPythonLinecache()

linecache.cache.bpython_history = bpython_history

# Monkey-patch the linecache module so that we're able

# to hold our command history there and have it persist

linecache.cache = BPythonLinecache(linecache.cache)

linecache.clearcache = _bpython_clear_linecache

def filename_for_console_input(code_string):

"""Remembers a string of source code, and returns

a fake filename to use to retrieve it later."""

try:

return linecache.cache.remember_bpython_input(code_string)

except AttributeError:

# If someone else has patched linecache.cache, better for code to

# simply be unavailable to inspect.getsource() than to raise

# an exception.

return '<input>'