Coverage for snippets/models.py: 100.00%

1from django.contrib.auth import get_user_model (empty)

2from django.db import models (empty)

3from pygments import highlight (empty)

4from pygments.formatters.html import HtmlFormatter (empty)

5from pygments.lexers import get_all_lexers, get_lexer_by_name (empty)

6from pygments.styles import get_all_styles (empty)

7 

8LEXERS = [item for item in get_all_lexers() if item[1]] (empty)

9LANGUAGE_CHOICES = sorted([(item[1][0], item[0]) for item in LEXERS]) (empty)

10STYLE_CHOICES = sorted((item, item) for item in get_all_styles()) (empty)

11 

12 

13class Snippet(models.Model): (empty)

14 created = models.DateTimeField(auto_now_add=True) (empty)

15 title = models.CharField(max_length=100, blank=True, default="") (empty)

16 code = models.TextField() (empty)

17 linenos = models.BooleanField(default=False) (empty)

18 language = models.CharField( (empty)

19 choices=LANGUAGE_CHOICES, default="python", max_length=100 

20 ) 

21 style = models.CharField(choices=STYLE_CHOICES, default="friendly", max_length=100) (empty)

22 owner = models.ForeignKey( (empty)

23 get_user_model(), related_name="snippets", on_delete=models.CASCADE 

24 ) 

25 highlighted = models.TextField() (empty)

26 

27 class Meta: (empty)

28 ordering = ("created",) (empty)

29 

30 def __str__(self): (empty)

31 return self.title 1 ctx1b

32 

33 def save(self, *args, **kwargs): (empty)

34 """ 

35 Use the 'pygments' library to create a highlighted HTML 

36 representation of the code snippet. 

37 """ 

38 lexer = get_lexer_by_name(self.language) (empty)

39 linenos = "table" if self.linenos else False (empty)

40 options = {"title": self.title} if self.title else {} (empty)

41 formatter = HtmlFormatter( (empty)

42 style=self.style, linenos=linenos, full=True, **options 

43 ) 

44 self.highlighted = highlight(self.code, lexer, formatter) (empty)

45 super().save(*args, **kwargs) (empty)