Hide keyboard shortcuts

Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

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

""" Dictionary-to-filetree functions, to create test files for testing. 

http://nedbatchelder.com/code/cog 

 

Copyright 2004-2019, Ned Batchelder. 

""" 

 

from __future__ import absolute_import 

 

import os.path 

 

from .backward import string_types, bytes_types 

from .whiteutils import reindentBlock 

 

__all__ = ['makeFiles', 'removeFiles'] 

 

def makeFiles(d, basedir='.', bytes=False): 

""" Create files from the dictionary `d`, in the directory named by `basedir`. 

If `bytes` is true, then treat bytestrings as bytes, else as text. 

""" 

for name, contents in d.items(): 

child = os.path.join(basedir, name) 

if isinstance(contents, string_types): 

mode = 'w' 

if bytes and isinstance(contents, bytes_types): 

mode += "b" 

f = open(child, mode) 

contents = reindentBlock(contents) 

f.write(contents) 

f.close() 

else: 

if not os.path.exists(child): 

os.mkdir(child) 

makeFiles(contents, child) 

 

def removeFiles(d, basedir='.'): 

""" Remove the files created by makeFiles. 

Directories are removed if they are empty. 

""" 

for name, contents in d.items(): 

child = os.path.join(basedir, name) 

if isinstance(contents, string_types): 

os.remove(child) 

else: 

removeFiles(contents, child) 

if not os.listdir(child): 

os.rmdir(child)