dfpyre 0.8.5__tar.gz → 0.8.6__tar.gz

{dfpyre-0.8.5 → dfpyre-0.8.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dfpyre
-Version: 0.8.5
+Version: 0.8.6
 Summary: A package for creating and modifying code templates for the DiamondFire Minecraft server.
 Home-page: https://github.com/Amp63/pyre
 License: MIT
@@ -12,7 +12,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: mcitemlib (>=0.3.4,<0.4.0)
+Requires-Dist: mcitemlib (>=0.4.1,<0.5.0)
 Project-URL: Repository, https://github.com/Amp63/pyre
 Description-Content-Type: text/markdown
 

{dfpyre-0.8.5 → dfpyre-0.8.6}/dfpyre/items.py

@@ -5,15 +5,19 @@ Class definitions for code items.
 from enum import Enum
 import re
 from typing import Literal, Any
-from dfpyre.style import is_ampersand_coded, ampersand_to_minimessage
-from dfpyre.util import PyreException, warn
+import websocket
 from mcitemlib.itemlib import Item as NbtItem, MCItemlibException
+from amulet_nbt import DoubleTag, StringTag, CompoundTag
+from dfpyre.style import is_ampersand_coded, ampersand_to_minimessage
+from dfpyre.util import PyreException, warn, COL_SUCCESS, COL_WARN, COL_ERROR, COL_RESET
 
 
 NUMBER_REGEX = r'^-?\d*\.?\d+$'
 VAR_SHORTHAND_REGEX = r'^\$([gsli]) (.+)$'
 VAR_SCOPES = {'g': 'unsaved', 's': 'saved', 'l': 'local', 'i': 'line'}
 
+CODECLIENT_URL = 'ws://localhost:31375'
+
 
 def convert_argument(arg: Any):
     if type(arg) in {int, float}:
@@ -99,7 +103,7 @@ class Item(NbtItem):
     type = 'item'
 
     def format(self, slot: int|None):
-        formatted_dict = {"item": {"id": self.type, "data": {"item": self.get_nbt()}}}
+        formatted_dict = {"item": {"id": self.type, "data": {"item": self.get_snbt()}}}
         _add_slot(formatted_dict, slot)
         return formatted_dict
 
@@ -111,50 +115,87 @@ class Item(NbtItem):
         Add a DiamondFire custom tag to this item.
         """
         if isinstance(tag_value, String):
-            tag_value = tag_value.value
+            tag = StringTag(tag_value.value)
+        elif isinstance(tag_value, str):
+            tag = StringTag(tag_value)
         elif isinstance(tag_value, Number):
-            tag_value = float(tag_value.value)
-        elif isinstance(tag_value, int):
-            tag_value = float(tag_value)
+            tag = DoubleTag(float(tag_value.value))
+        elif isinstance(tag_value, (int, float)):
+            tag = DoubleTag(float(tag_value))
         
         try:
-            item_tags = self.get_custom_data('PublicBukkitValues')
+            custom_data_tag = self.get_component('minecraft:custom_data')
+            if 'PublicBukkitValues' in custom_data_tag:
+                pbv_tag = custom_data_tag['PublicBukkitValues']
+            else:
+                pbv_tag = CompoundTag()
         except MCItemlibException:
-            item_tags = {}
+            custom_data_tag = CompoundTag()
+            pbv_tag = CompoundTag()
         
-        item_tags[f'hypercube:{tag_name}'] = tag_value
-        self.set_custom_data('PublicBukkitValues', item_tags)
+        custom_data_tag['PublicBukkitValues'] = pbv_tag
+        
+        pbv_tag[f'hypercube:{tag_name}'] = tag
+        self.set_component('minecraft:custom_data', custom_data_tag)
     
     def get_tag(self, tag_name: str) -> str|float|None:
         """
         Get a DiamondFire custom tag from this item.
         """
         try:
-            item_tags = self.get_custom_data('PublicBukkitValues')
+            custom_data_tag = self.get_component('minecraft:custom_data')
         except MCItemlibException:
             return None
         
-        try:
-            return item_tags[f'hypercube:{tag_name}']
-        except KeyError:
+        if 'PublicBukkitValues' not in custom_data_tag:
+            return None
+        
+        pbv_tag = custom_data_tag['PublicBukkitValues']
+        df_tag_value = pbv_tag.get(f'hypercube:{tag_name}')
+        if df_tag_value is None:
             return None
+        
+        if isinstance(df_tag_value, DoubleTag):
+            return float(df_tag_value)
+        if isinstance(df_tag_value, StringTag):
+            return str(df_tag_value)
     
-    def remove_tag(self, tag_name: str) -> bool:
+    def remove_tag(self, tag_name: str):
         """
         Remove a DiamondFire custom tag from this item.
-
-        :return: `True` on success, `False` on fail
         """
-        try:
-            item_tags = self.get_custom_data('PublicBukkitValues')
-        except MCItemlibException:
-            return False
+        custom_data_tag = self.get_component('minecraft:custom_data')
+        pbv_tag = custom_data_tag['PublicBukkitValues']
+        del pbv_tag[f'hypercube:{tag_name}']
 
+        return True
+    
+    def send_to_minecraft(self):
+        """
+        Sends this item to Minecraft automatically.
+        """
         try:
-            del item_tags[f'hypercube:{tag_name}']
-            return True
-        except KeyError:
-            return False
+            ws = websocket.WebSocket()
+            ws.connect(CODECLIENT_URL)
+            print(f'{COL_SUCCESS}Connected.{COL_RESET}')
+
+            command = f'give {self.get_snbt()}'
+            ws.send(command)
+            ws.close()
+
+            print(f'{COL_SUCCESS}Item sent to client successfully.{COL_RESET}')
+            return 0
+            
+        except Exception as e:
+            if isinstance(e, ConnectionRefusedError):
+                print(f'{COL_ERROR}Could not connect to CodeClient API. Possible problems:')
+                print(f'    - Minecraft is not open')
+                print(f'    - CodeClient is not installed (get it here: https://modrinth.com/mod/codeclient)')
+                print(f'    - CodeClient API is not enabled (enable it in CodeClient general settings)')
+                return 1
+            
+            print(f'Connection failed: {e}')
+            return 2
 
 
 class Location:
@@ -390,7 +431,7 @@ def item_from_dict(item_dict: dict) -> Any:
     item_data = item_dict['data']
 
     if item_id == 'item':
-        return Item.from_nbt(item_data['item'])
+        return Item.from_snbt(item_data['item'])
     
     elif item_id == 'txt':
         return String(item_data['name'])

{dfpyre-0.8.5 → dfpyre-0.8.6}/dfpyre/pyre.py

@@ -8,7 +8,7 @@ import json
 from difflib import get_close_matches
 import datetime
 from enum import Enum
-from mcitemlib.itemlib import Item as NbtItem
+from amulet_nbt import CompoundTag
 from dfpyre.util import *
 from dfpyre.items import *
 from dfpyre.scriptgen import generate_script, GeneratorFlags
@@ -162,7 +162,7 @@ def _check_applied_tags(tags: list[dict], applied_tags: dict[str, str], codebloc
 
 def _reformat_codeblock_tags(tags: list[dict], codeblock_type: str, codeblock_action: str, applied_tags: dict[str, str]):
     """
-    Turns tag objects into DiamondFire formatted tag items
+    Turns tag objects into DiamondFire formatted tag items.
     """
 
     def format_tag(option: str, name: str):
@@ -189,7 +189,7 @@ def _reformat_codeblock_tags(tags: list[dict], codeblock_type: str, codeblock_ac
 
 def _get_codeblock_tags(codeblock_type: str, codeblock_name: str, applied_tags: dict[str, str]):
     """
-    Get tags for the specified codeblock type and name
+    Get tags for the specified codeblock type and name.
     """
     action_data = CODEBLOCK_DATA[codeblock_type][codeblock_name]
     if 'deprecatedNote' in action_data:
@@ -198,10 +198,10 @@ def _get_codeblock_tags(codeblock_type: str, codeblock_name: str, applied_tags:
     return _reformat_codeblock_tags(tags, codeblock_type, codeblock_name, applied_tags)
 
 
-def _get_template_item(template_code: str, name: str, author: str) -> NbtItem:
+def _generate_template_item(template_code: str, name: str, author: str) -> Item:
     now = datetime.datetime.now()
 
-    template_item = NbtItem('yellow_shulker_box')
+    template_item = Item('yellow_shulker_box')
     template_item.set_name(f'&x&f&f&5&c&0&0>> &x&f&f&c&7&0&0{name}')
     template_item.set_lore([
         f'&8Author: {author}',
@@ -211,11 +211,13 @@ def _get_template_item(template_code: str, name: str, author: str) -> NbtItem:
         '&7https://github.com/Amp63/pyre'
     ])
     
-    pbv_tag = {
-        'hypercube:codetemplatedata': f'{{"author":"{author}","name":"{name}","version": 1,"code":"{template_code}"}}',
-        'hypercube:pyre_creation_timestamp': now.timestamp()
-    }
-    template_item.set_custom_data('PublicBukkitValues', pbv_tag, raw=True)
+    custom_data_tag = CompoundTag({
+        'PublicBukkitValues': CompoundTag({
+            'hypercube:codetemplatedata': StringTag(f'{{"author":"{author}","name":"{name}","version": 1,"code":"{template_code}"}}'),
+            'hypercube:pyre_creation_timestamp': DoubleTag(now.timestamp())
+        })
+    })
+    template_item.set_component('minecraft:custom_data', custom_data_tag)
 
     return template_item
 
@@ -316,15 +318,15 @@ class DFTemplate:
         return df_encode(json_string)
     
 
-    def build_and_send(self, method: Literal['recode', 'codeclient'], include_tags: bool=True) -> int:
+    def build_and_send(self, include_tags: bool=True) -> int:
         """
         Builds this template and sends it to DiamondFire automatically.
         
         :param bool include_tags: If True, include item tags in code blocks. Otherwise omit them.
         """
         template_code = self.build(include_tags)
-        template_item = _get_template_item(template_code, self._get_template_name(), self.author)
-        return template_item.send_to_minecraft(method, 'pyre')
+        template_item = _generate_template_item(template_code, self._get_template_name(), self.author)
+        return template_item.send_to_minecraft()
     
     
     def generate_script(self, output_path: str, indent_size: int=4, literal_shorthand: bool=True, var_shorthand: bool=False):

{dfpyre-0.8.5 → dfpyre-0.8.6}/dfpyre/scriptgen.py

@@ -40,13 +40,13 @@ class GeneratorFlags:
 
 
 def item_to_string(class_name: str, i: Item):
-    i.nbt.data.pop('~DF_NBT', None)
+    i.nbt.pop('DF_NBT', None)
     stripped_id = i.get_id().replace('minecraft:', '')
-    if i.nbt.key_set() == {'~id', '~count'}:
+    if set(i.nbt.keys()) == {'id', 'count'}:
         if i.get_count() == 1:
             return f'{class_name}("{stripped_id}")'
         return f'{class_name}("{stripped_id}", {i.get_count()})'
-    return f'{class_name}.from_nbt("""{i.get_nbt()}""")'
+    return f'{class_name}.from_snbt("""{i.get_snbt()}""")'
 
 
 def argument_item_to_string(flags: GeneratorFlags, arg_item: object) -> str:
@@ -56,12 +56,12 @@ def argument_item_to_string(flags: GeneratorFlags, arg_item: object) -> str:
     
     if isinstance(arg_item, String):
         value = arg_item.value.replace('\n', '\\n')
+        if flags.literal_shorthand:
+            return f'"{value}"'
         return f'{class_name}("{value}")'
     
     if isinstance(arg_item, Text):
         value = arg_item.value.replace('\n', '\\n')
-        if flags.literal_shorthand:
-            return f'"{value}"'
         return f'{class_name}("{value}")'
     
     if isinstance(arg_item, Number):

{dfpyre-0.8.5 → dfpyre-0.8.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dfpyre"
-version = "0.8.5"
+version = "0.8.6"
 description = "A package for creating and modifying code templates for the DiamondFire Minecraft server."
 authors = ["Amp"]
 readme = "README.md"
@@ -12,7 +12,7 @@ exclude = ['examples', 'extra']
 
 [tool.poetry.dependencies]
 python = "^3.10"
-mcitemlib = "^0.3.4"
+mcitemlib = "^0.4.1"
 
 
 [build-system]