FilesExpand file tree

 master

/

files.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

79 lines (67 loc) · 2.98 KB

 master

/

files.py

Copy path

Top

File metadata and controls

• Code
• Blame

79 lines (67 loc) · 2.98 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

76

77

78

79

# -*- coding: UTF-8 -*-

# Copyright © 2010 Piotr Ożarowski <piotr@debian.org>

#

# Permission is hereby granted, free of charge, to any person obtaining a copy

# of this software and associated documentation files (the "Software"), to deal

# in the Software without restriction, including without limitation the rights

# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

# copies of the Software, and to permit persons to whom the Software is

# furnished to do so, subject to the following conditions:

#

# The above copyright notice and this permission notice shall be included in

# all copies or substantial portions of the Software.

#

# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

# THE SOFTWARE.

import logging

from os import walk

from os.path import abspath, isfile, join

from subprocess import Popen, PIPE

log = logging.getLogger(__name__)

def from_directory(dname, extensions=('.py',)):

"""Generate *.py file names available in given directory."""

extensions = tuple(extensions) # .endswith doesn't like list

if isinstance(dname, (list, tuple)):

for item in dname:

for fn in from_directory(item):

yield fn

elif isfile(dname) and dname.endswith(extensions):

yield dname

else:

for root, dirs, file_names in walk(abspath(dname)):

for fn in file_names:

if fn.endswith(extensions):

yield join(root, fn)

def from_package(package_name, extensions=('.py',)):

"""Generate *.py file names available in given package."""

extensions = tuple(extensions) # .endswith doesn't like list

process = Popen("/usr/bin/dpkg -L %s" % package_name,\

shell=True, stdout=PIPE)

stdout, stderr = process.communicate()

if process.returncode != 0:

raise Exception("cannot get content of %s" % package_name)

for line in stdout.splitlines():

if line.endswith(extensions):

yield line

def filter_directory(files, dname):

"""Generate *.py file names that match given directory."""

for fn in files:

if fn.startswith(dname):

yield fn

def filter_public(files, versions):

"""Generate *.py file names that match given versions."""

versions_str = set("%d.%d" % i for i in versions)

for fn in files:

if fn.startswith('/usr/lib/python') and \

fn[15:18] in versions_str:

yield fn

def filter_out_ext(files, extensions):

"""Removes files with matching extensions from given generator."""

for fn in files:

if not fn.endswith(extensions):

yield fn